Leetcode802-找到最终的安全状态(Python3)

刚开始没思路，还以为是利用二维矩阵直接标记节点间的有向路径，最后循环遍历就能得到结果，结果最后发现方向是错的，之后看了大佬们写的代码，发现原来是用出度来实现节点是否安全的。

照着大佬们的思路重新写了一遍，代码如下：

 class Solution:
     def eventualSafeNodes(self, graph: List[List[int]]) -> List[int]:
         #节点个数N
         N=len(graph)
         #safe数组记录N个节点是否安全，1表示安全，0表示不安全
         safe=[0 for _ in range(N)]
         #记录每个节点的出度，出度为0的节点安全
         outDegree=[0 for _ in range(N)]
         #逆邻接链表，就是有向边进行反向
         reverseGraph=[[] for _ in range(N)]

         #计算出度并构建逆邻接链表
         for i in range(N):
             outDegree[i] = len(graph[i])#出度计算
             for j in graph[i]:#反向边
                 reverseGraph[j].append(i)

         #统计出度为0的节点
         safeQueue=[]
         for i in range(N):
             if outDegree[i]==0:
                 safeQueue.append(i)
         #对出度为0的节点进行反向查找
         while safeQueue:
             node = safeQueue.pop(0)
             safe[node]=1
             for i in reverseGraph[node]:
              outDegree[i]-=1
              if outDegree[i]==0:
                  safeQueue.append(i)
         #统计安全节点
         res=[]
         for i in range (N):
             if safe[i]==1:
                 res.append(i)
         return res

当然，利用dfs查重也是可以的，但是太慢了，就不考虑了