1. 程式人生 > 其它 >684. 冗餘連線

684. 冗餘連線

技術標籤:並查集資料結構與演算法leetcode

題目描述:在本問題中, 樹指的是一個連通且無環的無向圖。
輸入一個圖,該圖由一個有著N個節點 (節點值不重複1, 2, …, N) 的樹及一條附加的邊構成。附加的邊的兩個頂點包含在1到N中間,這條附加的邊不屬於樹中已存在的邊。
結果圖是一個以邊組成的二維陣列。每一個邊的元素是一對[u, v] ,滿足 u < v,表示連線頂點u 和v的無向圖的邊。
返回一條可以刪去的邊,使得結果圖是一個有著N個節點的樹。如果有多個答案,則返回二維陣列中最後出現的邊。答案邊 [u, v] 應滿足相同的格式 u < v。
解題思路:無向無環圖多增加了一條邊,那必然會形成環路,利用並查集,依次遍歷題目給出的邊,如果當前兩個節點所屬連通域不同就合併當前兩個集合,如果他們已經聯通,因為當前這條新增的邊導致形成環路,因此當前這條邊作為候選結果,如果之後還有這種情況,更新返回結果,程式碼如下:

class Solution:
    def findRedundantConnection(self, edges: List[List[int]]) -> List[int]:
        n = len(edges)
        father = list(range(n+1))
        res = []
        
        def find(v):
            root = v
            while(root != father[root]):
                root = father[root]
            while
(v != root): origin_root = father[v] father[v] = root v = origin_root return root def merge(a, b): root_a, root_b = find(a), find(b) if root_a != root_b: father[root_b] = root_a for e in
edges: if find(e[0]) == find(e[1]): res = e else: merge(e[0], e[1]) return res