LeetCode 802 找到最终的安全状态[DFS 三色标记法] HERODING的LeetCode之路

Posted 2021-09-01 HERODING23

解题思路：
这是一道读题特别费脑子的题目，大致意思是这样，给你一排点，从0开始按序标号，并且给你每个点的下一访问点的集合，问从某一点出发，沿任意路径走最后都能走到尽头（而不是死循环）的点的集合是多少。
希望我的描述不会给你带来更多的困惑，这题使用的方法是比较少见的三色标记法，代表着每个点都有着三种状态，第一个是未访问，第二个是已访问，第三个是安全状态，一开始每个点都是未访问，在访问过程中，如果遇到了访问过的点，那就代表着要走死循环的路子了，直接返回false，如果访问完成都到达终点（即没有遇到访问过的点），那么标记安全状态，返回true，代码如下：

class Solution {
public:
    vector<int> eventualSafeNodes(vector<vector<int>>& graph) {
        int n = graph.size();
        // 定义标记
        vector<int> mark(n);

        // 自定义函数，用于判断该点是否是安全节点
        function<bool(int)> judge = [&](int x) {
            // 有标记，观察是否是走过还是安全点
            if(mark[x] > 0) {
                return mark[x] == 2;
            }
            // 先标记为走过
            mark[x] = 1;
            // 遍历下一个节点，观察是否会走过
            for(int& i: graph[x]) {
                if(!judge(i)) {
                    return false;
                }
            }
            // 下一节点不是走过状态，即是安全的
            mark[x] = 2;
            return true;
        };

        vector<int> ans;
        for(int i = 0; i < n; i ++) {
            if(judge(i)) {
                ans.push_back(i);
            }
        }
        return ans;
    }
};