684. 冗餘連線
阿新 • • 發佈:2021-01-14
題目描述:在本問題中, 樹指的是一個連通且無環的無向圖。
輸入一個圖,該圖由一個有著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