LeeCode 802 dfs

最新推荐文章于 2022-05-20 21:21:01 发布

SHOHOKUKU

最新推荐文章于 2022-05-20 21:21:01 发布

阅读量104

点赞数

分类专栏：搜索

本文链接：https://blog.csdn.net/neweryyy/article/details/106674879

版权

搜索专栏收录该内容

58 篇文章 0 订阅

订阅专栏

题意

传送门 LeeCode 802. 找到最终的安全状态

题解

有向图中，若节点能走到圈中，则不是安全节点；反之，则是安全节点。那么 $d f s$ 时，除了维护访问过的节点，同时维护节点是否为安全节点，将不能走到圈的节点加入答案即可。

class Solution
{
public:
    vector<int> res;
    vector<bool> used, label;
    bool dfs(int u, vector<vector<int>> &G)
    {
        used[u] = true;
        bool flag = true;
        for (int i = 0; i < G[u].size(); i++)
        {
            int w = G[u][i];
            flag &= (used[w] ? label[w] : dfs(w, G));
        }
        if (flag)
        {
            res.push_back(u);
            label[u] = true;
        }
        return flag;
    }
    vector<int> eventualSafeNodes(vector<vector<int>> &graph)
    {
        int v = graph.size();
        used.resize(v, false);
        label.resize(v, false);
        for (int i = 0; i < v; i++)
        {
            if (!used[i])
            {
                dfs(i, graph);
            }
        }
        sort(res.begin(), res.end());
        return res;
    }
};