码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode 802. 找到最终的安全状态(Find Eventual Safe States)

时间:2019-05-24 10:49:09      阅读:155      评论:0      收藏:0      [点我收藏+]

标签:order   href   etc   http   back   ref   pre   vector   没有   

题目描述:

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

现在, 如果我们最后能走到终点,那么我们的起始节点是最终安全的。 更具体地说, 存在一个自然数 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]
这里是上图的示意图。

技术图片

提示:

  • graph 节点数不超过 10000.
  • 图的边数不会超过 32000.
  • 每个 graph[i] 被排序为不同的整数列表, 在区间 [0, graph.length - 1] 中选取。

解法:

class Solution {
public:
    vector<int> eventualSafeNodes(vector<vector<int>>& graph) {
        vector<int> res;
        int sz = graph.size();
        vector<int> degree(sz, 0);
        unordered_map<int, vector<int>> redge;
        for(int i=0; i<sz; i++){
            redge[i] = {};
        }
        stack<int> leaf;
        for(int i=0; i<sz; i++){
            degree[i] += graph[i].size();
            if(degree[i] == 0){
                leaf.push(i);
            }
            for(int val : graph[i]){
                redge[val].push_back(i);
            }
        }
        while(!leaf.empty()){
            int node = leaf.top();
            leaf.pop();
            res.push_back(node);
            for(int nxt : redge[node]){
                degree[nxt]--;
                if(degree[nxt] == 0){
                    leaf.push(nxt);
                }
            }
        }
        sort(res.begin(), res.end());
        return res;
    }
};

leetcode 802. 找到最终的安全状态(Find Eventual Safe States)

标签:order   href   etc   http   back   ref   pre   vector   没有   

原文地址:https://www.cnblogs.com/zhanzq/p/10916472.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!