Leetcode 207. 课程表

最新推荐文章于 2024-04-07 01:18:54 发布

原创最新推荐文章于 2024-04-07 01:18:54 发布 · 287 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #拓扑排序 #深搜 #广搜

LeetCode 专栏收录该内容

210 篇文章

订阅专栏

该博客讨论了LeetCode上的207号问题——课程表，介绍了如何判断是否能完成所有课程的学习。通过一个示例展示了当有先修课程条件时，如果要学习课程0，则必须先完成课程1。文章提供了C++语言的解题思路，包括拓扑排序的广搜实现方法，并鼓励读者尝试解决并尊重版权信息。

题目描述
你这个学期必须选修 numCourses 门课程，记为 0 到 numCourses - 1 。

在选修某些课程之前需要一些先修课程。先修课程按数组 prerequisites 给出，其中 prerequisites[i] = [ai, bi] ，表示如果要学习课程 ai 则必须先学习课程 bi 。

例如，先修课程对 [0, 1] 表示：想要学习课程 0 ，你需要先完成课程 1 。
请你判断是否可能完成所有课程的学习？如果可以，返回 true ；否则，返回 false 。

示例 1：

输入：numCourses = 2, prerequisites = [[1,0]]
输出：true
解释：总共有 2 门课程。学习课程 1 之前，你需要完成课程 0 。这是可能的。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/course-schedule
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
C++

方法一：

class Solution {
    /*统计每个课被指向次数，初始被指向次数为0的肯定是安全的（不在环上）。

每被安全课程指向一次，被指次数减一，

如果被指次数减到0，说明该课程全部指向都来自安全课程，则它也是安全的。

依此进行队列循环。*/
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        int len=prerequisites.size();
        if(!len)  return true;
        vector<int> precount(numCourses); //记录每个课程被指向的次数
        for(int i=0;i<prerequisites.size();i++)
        {
            precount[prerequisites[i][0]]++;
        }

        vector<bool> isremoved(len); //标记课程是否安全了
        int removed=0;
        while(removed<len){
            int cur_removed=0; //标记本轮移除的元素数目,这里的元素值得是课程对
            for(int i=0;i<len;i++){
                if(isremoved[i])  continue;
                if(precount[ prerequisites[i][1]]==0) {
                    precount[ prerequisites[i][0]]--;
                    //被安全课程指向，被指次数就减一
                    isremoved[i]=true;
                    ++cur_removed;
                }
            }
            if(cur_removed==0)  return false;//一轮一个元素都没移除，说明成环了
            removed+=cur_removed;
        }
        return true;
    }
};

方法二：
广搜实现拓扑排序

参看题解

class Solution {
    /*
    广度优先搜索，实现拓扑排序
    */
    vector<vector<int>> edge;
    vector<int> indge;  //入度
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        edge.resize(numCourses);
        indge.resize(numCourses);
        for(int i=0;i<prerequisites.size();i++){
            edge[prerequisites[i][1]].push_back(prerequisites[i][0]);
            ++indge[prerequisites[i][0]];
        }

        queue<int> q;
        //初始化队列
        for(int i=0;i<numCourses;i++){
            if(indge[i]==0)
                q.push(i);
        }
        //遍历
        int visited=0;
        while(!q.empty()){
            ++visited;
            int index=q.front();
            q.pop();
            for(int v : edge[index]){
                indge[v]--;
                if(indge[v]==0){
                    q.push(v);
                }
            }
        }
        return visited==numCourses;
    }
};