Leetcode 207 课程表 (拓扑排序,判断有向图环)

定义一个队列Q,把入度为0的结点入队

若Q不为空,则取队首结点,删去所有从该点出发的边,并把这些边所到达结点的入度减一,若某个节点入度减为0,则将它入队

反复进行如上操作,直到队列为空。(当总的入队次数大于节点数时,跳出循环)

如果这时入过队的节点数恰好等于节点总数,则为有向无环图。否则有环。

 

class Solution {
public:
    static const int maxn = 10000;
    vector<int> gra[maxn];
    int co = 0;
    int count[maxn] = { 0 };  //入度
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        queue<int> Q;
        for (int i = 0;i < prerequisites.size();i++) {
            gra[prerequisites[i][1]].push_back(prerequisites[i][0]);
            count[prerequisites[i][0]]++;
        }
        int co = 0;
        for (int i = 0;i < numCourses;i++)
            if (count[i]==0) {
                Q.push(i);
            }
        while (!Q.empty()) {
            int cur = Q.front();
            Q.pop();
            co++;
            if (co > numCourses)break;
            for (int i = 0;i < gra[cur].size();i++) {
                count[gra[cur][i]]--;
                if (count[gra[cur][i]] == 0) {
                    Q.push(gra[cur][i]);
                }
            }
            gra[cur].clear();
        }
        if (co == numCourses)return true;
        else return false;
    }
};

 

posted @ 2019-06-20 08:41  suuusu  阅读(344)  评论(0编辑  收藏  举报