不知這裡有沒有高手有參加這週的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) 感謝各位願意看完 --

※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 108.254.89.199 (美國) ※ 文章網址: https://webptt.com/m.aspx?n=bbs/Python/M.1638136732.A.669.html