LeetCode #802

Find Eventual Safe States

1개의 풀이 · C++

문제 원문 보기 ↗

SOLUTION INFO

C++ · main.cpp

main.cpp
class Solution {
public:
    vector<int> eventualSafeNodes(vector<vector<int>>& graph) {
        int N = (int)graph.size();
        vector<vector<int>> rG(N);
        vector<int> ind(N);
        for(int i = 0; i < N; ++i) {
            for(int prv: graph[i]) {
                rG[prv].push_back(i);
                ++ ind[i];
            }
        }
        queue<int> Q;
        vector<int> chk(N);
        for(int i = 0; i < N; ++i) {
            if(ind[i] == 0) Q.push(i), chk[i] =  1;
        }
        while(!Q.empty()) {
            int cur = Q.front(); Q.pop();
            for(int nxt: rG[cur]) {
                if(--ind[nxt] == 0) Q.push(nxt), chk[nxt] |= chk[cur];
            }
        }
        vector<int> answer;
        for(int i = 0; i < N; ++i) {
            if(chk[i]) answer.push_back(i);
        }
        return answer;
    }
};

SOLUTION DESCRIPTION

풀이 설명

등록된 풀이 설명이 없습니다.