LeetCode 3310. 移除可疑的方法

你正在维护一个项目，该项目有 n 个方法，编号从 0 到 n - 1。
给你两个整数 n 和 k，以及一个二维整数数组 invocations，其中 invocations[i] = [ai, bi] 表示方法 ai 调用了方法 bi。
已知如果方法 k 存在一个已知的 bug。那么方法 k 以及它直接或间接调用的任何方法都被视为 可疑方法 ，我们需要从项目中移除这些方法。
只有当一组方法没有被这组之外的任何方法调用时，这组方法才能被移除。
返回一个数组，包含移除所有 可疑方法 后剩下的所有方法。你可以以任意顺序返回答案。如果无法移除 所有 可疑方法，则 不 移除任何方法。
示例 1:
输入: n = 4, k = 1, invocations = [[1,2],[0,1],[3,2]]
输出: [0,1,2,3]
解释:

方法 2 和方法 1 是可疑方法，但它们分别直接被方法 3 和方法 0 调用。由于方法 3 和方法 0 不是可疑方法，我们无法移除任何方法，故返回所有方法。
示例 2:

输入: n = 5, k = 0, invocations = [[1,2],[0,2],[0,1],[3,4]]
输出: [3,4]
解释:
方法 0、方法 1 和方法 2 是可疑方法，且没有被任何其他方法直接调用。我们可以移除它们。
示例 3:

输入: n = 3, k = 2, invocations = [[1,2],[0,1],[2,0]]
输出: []
解释:
所有方法都是可疑方法。我们可以移除它们。
提示:
1 <= n <= 105
0 <= k <= n - 1
0 <= invocations.length <= 2 * 105
invocations[i] == [ai, bi]
0 <= ai, bi <= n - 1
ai != bi
invocations[i] != invocations[j]

图、图的遍历

class Node:def __init__(self, val):self.val = valself.next = []self.prev = []class Solution:def remainingMethods(self, n: int, k: int, invocations: List[List[int]]) -> List[int]:if not invocations:return list(set(range(n)) - set([k]))# 建图method_mapping = {}for a, b in invocations:# a -> bif a not in method_mapping:method_mapping[a] = Node(a)if b not in method_mapping:method_mapping[b] = Node(b)node = method_mapping[a]node.next.append(method_mapping[b])for i in range(n):if i not in method_mapping:method_mapping[i] = Node(i)# 查出所有可疑方法bad_method_set = set()q = deque([method_mapping[k]])while q:node = q.popleft()bad_method_set.add(node.val)for i in node.next:if i.val not in bad_method_set:q.append(i)# 验证是否刚好能移除可疑方法flag = Truefor a, b in invocations:if a in bad_method_set and b not in bad_method_set:flag = Falsebreakelif a not in bad_method_set and b in bad_method_set:flag = Falsebreakl = bad_method_set if flag else set()return list(set(range(n)) - l)