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

是小Y啦2024-05-29 8:53

思路：拓补排序

其实这道题只要把顺序倒过来就行了，我们首先看到没有出度的反而是终端点，我们不如让它反过来成为没有入度的点是终端店，这样的话我们用度的个数来找到终端点就很容易了。

那么这样的话，题目中说若点满足所有的路径都到达终端点，那么这个点就是安全点

其实我们反过来说的话其实就是从所有的终端点出发，能不能都到达这个点，那么这个点就是满足安全点的条件的。

我们也发现了，求解这种多终点的输出路径问题，其实可以用拓补排序来做。接下来其实就是拓补排序了，用一个队列来存储路径点就可以了，最后排序即可。

复制代码

class Solution {
public:
    vector<int> eventualSafeNodes(vector<vector<int>>& graph) {
        vector<vector<int>>s(graph.size());
        vector<int>counts(graph.size(),0);
        for(int i=0;i<graph.size();i++){
            for(int j=0;j<graph[i].size();j++){
                s[graph[i][j]].push_back(i);
            }
            counts[i]+=graph[i].size();
        }
        vector<int>ans;//终端节点
        vector<int>res;
        queue<int>q;
        for(int i=0;i<graph.size();i++){
            if(counts[i]==0){
                ans.push_back(i); 
                res.push_back(i);
                q.push(i);            
            }
        }
        while(!q.empty()){
            int tmp=q.front();
            q.pop();
            for(int i=0;i<s[tmp].size();i++){
                if(--counts[s[tmp][i]]==0){
                    q.push(s[tmp][i]);
                    res.push_back(s[tmp][i]);
                }
            }
        }
        sort(res.begin(),res.end());
        return res;
        

    } 
};