LeetCode 802.找到最终的安全状态

LeetCode 802.找到最终的安全状态

在有向图中, 我们从某个节点和每个转向处开始, 沿着图的有向边走。 如果我们到达的节点是终点 (即它没有连出的有向边), 我们停止。

现在, 如果我们最后能走到终点，那么我们的起始节点是最终安全的。 更具体地说, 存在一个自然数 K,  无论选择从哪里开始行走, 我们走了不到 K 步后必能停止在一个终点。

哪些节点最终是安全的？ 结果返回一个有序的数组。

该有向图有 N 个节点，标签为 0, 1, ..., N-1, 其中 N 是 graph 的节点数.  图以以下的形式给出: graph[i] 是节点 j 的一个列表，满足 (i, j) 是图的一条有向边。

示例：
输入：graph = [[1,2],[2,3],[5],[0],[5],[],[]]
输出：[2,4,5,6]
这里是上图的示意图。

看起来很乱，其实要求就是无论选择从哪里开始行走最终都能走到终点

平时都是任意有一条能到达就行，这题是要求所有的边都能到达，挺新鲜

样例中的0->1,1->3,3->1这样形成的循环，走不到终点，所以这道题就是判断每个点的所有情况能不能都走到终点

用vis数组判断是否被访问过，safe数组判断该结点是否为安全节点

class Solution
{
  private:
    static const int maxn = 32005;
    bool vis[maxn], safe[maxn];
    bool dfs(int cur, vector<vector<int>> &graph)
    {
    	//先判断是否为安全节点，因为是深度优先，所以可能会
    	//后访问到一个安全结点，而该结点先前一定被访问过了
    	//比如把样例改成[[1,2],[2,3],[5],[6],[5],[],[]]
    	//不先判断安全的话0结点不会加入答案
        if (safe[cur])
            return true;
        //不安全并且访问过，那一定是不可以的
        if (vis[cur])
            return false;
        vis[cur] = true;
        
        if (graph[cur].size() == 0)//终点，设为安全
        {
            safe[cur] = true;
            return true;
        }
        
        for (int i = 0; i < graph[cur].size(); ++i)
            if (!dfs(graph[cur][i], graph))//任意一边不满足返回假
                return false;
        safe[cur] = true;//记得把当前结点设为安全
        return true;
    }

  public:
    vector<int> eventualSafeNodes(vector<vector<int>> &graph)
    {
        vector<int> ans;
        memset(safe, 0, sizeof(safe));
		
        for (int i = 0; i < graph.size(); ++i)
        {
            memset(vis, 0, sizeof(vis));
            if (dfs(i, graph))
                ans.push_back(i);
        }
        return ans;
    }
};