不知這裡有沒有高手有參加這週的Leetcode 週賽，想請教leetcode 2029
https://leetcode.com/problems/find-all-people-with-secret/
這題我是用Union-Find來做的，思路大致是：
先用一個dictionary把同一個時間的meeting放在一起，然後由時間小的loop到時間大的
如果該meeting中的參與人x, y中有一個和firstPerson是同一個根節點，則union
在每一個union操作後，將x, y皆放入res
同個時間若有多個meeting，則用一個while loop，不斷檢查該時間的所有x, y組合
直至res不再變動
以下是我的code，我一直想不透錯在哪，到第38個test case時fail了
class Solution(object):
def findAllPeople(self, n, meetings, firstPerson):
"""
:type n: int
:type meetings: List[List[int]]
:type firstPerson: int
:rtype: List[int]
"""
parent = {i: i for i in range(n)}
res = set()
res.add(0)
res.add(firstPerson)
def find(x):
if x != parent[x]:
parent[x] = find(parent[x])
return parent[x]
def union(a, b):
pa, pb = find(a), find(b)
if pa!=pb:
parent[pa] = pb
union(0, firstPerson)
hmap = collections.defaultdict(list)
for a, b, time in meetings:
hmap[time].append((a, b))
for time in sorted(hmap.keys()):
arr = hmap[time]
while True:
tmp = res
for a, b in arr:
if find(a) == find(firstPerson) or find(b) ==
find(firstPerson):
union(a,b)
res.add(a)
res.add(b)
if tmp == res:
break
return list(res)
感謝各位願意看完