其他分享
首页 > 其他分享> > LeetCode 207 Course Schedule 拓扑排序BFS

LeetCode 207 Course Schedule 拓扑排序BFS

作者:互联网

There are a total of numCourses courses you have to take, labeled from 0 to numCourses - 1. You are given an array prerequisites where prerequisites[i] = [ai, bi] indicates that you must take course bi first if you want to take course ai.

For example, the pair [0, 1], indicates that to take course 0 you have to first take course 1.
Return true if you can finish all courses. Otherwise, return false.

Solution

拓扑排序的模板题。每次建边从 \(b_i \rightarrow a_i\),然后 \(a_i\) 的度数 \(+1\)。每次取出度数为 \(0\) 的点加入队列,然后将删除这些点相连的边:即将 \(a_i\) 的度数 \(-1\),再将度数为 \(0\) 的点加入队列直到没有点满足条件。如果最后符合条件点数 \(=\ \text{numCourses}\),则为true;否则为false

点击查看代码
class Solution {
private:
    vector<int> vec[2004];
    int deg[2002];
    vector<int> ans;
public:
    bool canFinish(int numCourses, vector<vector<int>>& preq) {
        for(int i=0;i<preq.size();i++){
            vec[preq[i][1]].push_back(preq[i][0]);
            deg[preq[i][0]]+=1;
        }
        for(int i=0;i<numCourses;i++){
            if(deg[i]==0)ans.push_back(i);
        }
        for(int i=0;i<ans.size();i++){
            int N = vec[ans[i]].size();
            for(int j=0;j<N;j++){
                deg[vec[ans[i]][j]]-=1;
                if(deg[vec[ans[i]][j]]==0)ans.push_back(vec[ans[i]][j]);
            }
        }
        if(ans.size()==numCourses)return true;
        return false;
    }
};

标签:度数,207,Schedule,int,numCourses,BFS,course,vector,take
来源: https://www.cnblogs.com/xinyu04/p/16283142.html