溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

python實(shí)現(xiàn)不鄰接植花的方法

發(fā)布時(shí)間:2020-07-29 10:38:09 來(lái)源:億速云 閱讀:131 作者:小豬 欄目:開(kāi)發(fā)技術(shù)

這篇文章主要講解了python實(shí)現(xiàn)不鄰接植花的方法,內(nèi)容清晰明了,對(duì)此有興趣的小伙伴可以學(xué)習(xí)一下,相信大家閱讀完之后會(huì)有幫助。

有 N 個(gè)花園,按從 1 到 N 標(biāo)記。在每個(gè)花園中,你打算種下四種花之一。

paths[i] = [x, y] 描述了花園 x 到花園 y 的雙向路徑。

另外,沒(méi)有花園有 3 條以上的路徑可以進(jìn)入或者離開(kāi)。

你需要為每個(gè)花園選擇一種花,使得通過(guò)路徑相連的任何兩個(gè)花園中的花的種類(lèi)互不相同。

以數(shù)組形式返回選擇的方案作為答案 answer,其中 answer[i] 為在第 (i+1) 個(gè)花園中種植的花的種類(lèi)?;ǖ姆N類(lèi)用 1, 2, 3, 4 表示。保證存在答案。

示例 1:

輸入:N = 3, paths = [[1,2],[2,3],[3,1]]

輸出:[1,2,3]

示例 2:

輸入:N = 4, paths = [[1,2],[3,4]]

輸出:[1,2,1,2]

示例 3:

輸入:N = 4, paths = [[1,2],[2,3],[3,4],[4,1],[1,3],[2,4]]

輸出:[1,2,3,4]

提示:

1 <= N <= 10000
0 <= paths.size <= 20000

不存在花園有 4 條或者更多路徑可以進(jìn)入或離開(kāi)。
保證存在答案。

知識(shí)準(zhǔn)備

在python中可以使用列表作為隊(duì)列,list用append添加元素

可以用字典來(lái)存儲(chǔ)鄰接節(jié)點(diǎn)nei = {}

在集合中使用for循環(huán)

{res[j] for j in G[i]}

集合的pop函數(shù)

flowers = {1,2,3,4} #集合直接相減即可
flowers.pop()
# 集合不能獲取某個(gè)元素這樣子的操作
print(flowers)

out: {2,3,4}集合中的pop是從左邊開(kāi)始取

集合的相減

flowers = {1,2,3,4}
h = {0}
flowers-h

out:{1,2,3,4}

我的題解

題解1

 
 class Solution:
   # 整體思路采用BFS方法,還需考慮不連通圖的問(wèn)題,然后著手結(jié)果唯一
   def gardenNoAdj(self, N: int, paths: List[List[int]]) -> List[int]:
     #構(gòu)建一個(gè)answer數(shù)組
     answer = [0 for _ in range(N)]
     #構(gòu)建所有節(jié)點(diǎn)
     all_nodes = []
     [all_nodes.append(i) for i in range(1,N+1)]
     #構(gòu)建visted列表
     visted = dict.fromkeys(all_nodes, 0)
     #初始化nei字典元素為空列表
     nei = [[] for _ in range(N)]
     # 構(gòu)建無(wú)向鄰接表,無(wú)鄰居則不構(gòu)建
     for path in paths:
       nei[path[0]-1].append(path[1])
       nei[path[1]-1].append(path[0])
     #遍歷每一個(gè)點(diǎn),每個(gè)點(diǎn)保證自己鄰接點(diǎn)不是和自己相同就行
     answer[0] = 1 
     for node in range(1,N+1):  #遍歷所有節(jié)點(diǎn)
       visted[node] = 1
       fix = set()
       if(answer[node-1]==0): #如果為0,說(shuō)明不是連通圖
         answer[node-1] = 1 
       flowers=[1,2,3,4]
       nei[node-1] = sorted(nei[node-1]) #排序鄰居節(jié)點(diǎn)
       flowers.pop(answer[node-1]-1) #彈出父節(jié)點(diǎn)的flowers
       for sinode in nei[node-1]: #遍歷鄰居
         if(visted[sinode] == 0): #如果鄰居未被訪問(wèn)過(guò)
           answer[sinode-1] = flowers[0] #使用1,彈出1
           flowers.pop(0)
         else: #如果鄰居被訪問(wèn)過(guò)
           if(answer[sinode-1]==answer[node-1]):
             answer[node-1] = flowers[0] 
             flowers.pop(0) 
           fix.add(answer[sinode-1])
       if not fix:
         continue
       else:
         flowers=[1,2,3,4]
         for a_val in list(fix):
           flowers.remove(a_val)
         answer[node-1] = flowers[0]
             
     return answer 

簡(jiǎn)化方法:利用集合快速搞定

class Solution:
  def gardenNoAdj(self, N: int, paths: List[List[int]]) -> List[int]:
   #構(gòu)建一個(gè)answer數(shù)組
    answer = [0]*N
    #初始化nei字典元素為空列表
    nei = [[] for _ in range(N)]
    # 構(gòu)建無(wú)向鄰接表,無(wú)鄰居則不構(gòu)建
    for path in paths:
      nei[path[0]-1].append(path[1])
      nei[path[1]-1].append(path[0])
    for node in range(1,N+1):  #遍歷所有節(jié)點(diǎn)
      flowers={1,2,3,4}
      #臨時(shí)存儲(chǔ)鄰居含有的花類(lèi)型
      a = set()
      for sinode in nei[node-1]: #遍歷鄰居
        a.add(answer[sinode-1])
      flowers = flowers - a 
      answer[node-1] = flowers.pop()
                
    return answer

看完上述內(nèi)容,是不是對(duì)python實(shí)現(xiàn)不鄰接植花的方法有進(jìn)一步的了解,如果還想學(xué)習(xí)更多內(nèi)容,歡迎關(guān)注億速云行業(yè)資訊頻道。

向AI問(wèn)一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI