拓扑排序模板【广度优先搜索】

leetcode链接:

207. 课程表

一、题目描述:

        你这个学期必须选修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。这是可能的。

提示:

  • 1 <= numCourses <= 2000
  • 0 <= prerequisites.length <= 5000
  • prerequisites[i].length == 2
  • 0 <= ai, bi < numCourses
  • prerequisites[i] 中的所有课程对 互不相同

二、思路:

        考虑拓扑排序中最前面的节点,该节点一定没有任何入边,即没有任何的先修课程要求;

        当将一个节点加入答案后,就可以移除它的所有出边,代表着它的相邻节点少了一门先修课程的要求;

        如果某个相邻节点变成没有任何入边的节点,那么就代表这门课可以开始学习了;

        按照这种流程,不断将没有入边的节点加入答案,直到答案中包含所有的节点【拓扑排序】或者不存在没有入边的节点【有环】。

三、代码:

class Solution {
public:
    bool canFinish(int numCourses, vector>& prerequisites) {
        edges.resize(numCourses);
        indeg.resize(numCourses);
        for(auto info : prerequisites) {
            edges[info[1]].push_back(info[0]);
            ++indeg[info[0]];
        }

        queue q;
        for(int i=0;i> edges;
    vector indeg;
};

四、题目拓展:

        小米手机的生产制造过程依赖复杂的焊接,集成,封装,测试等流程,工序之间存在相互依赖,假设必须完成的 n 门工序,记为0到n-1,在完成某些工序之前需要先完成前置工序。其中[a:b],表示如果要完成工序a,必须先完成工序b,请你判断工序的编排是否合理,是否可能完成所有工序?如果可以,输出1,否则,输出0.

        样例:

                输入:2

                           1:0,0:1

                输出:0

注意:ACM模式下,考虑输入格式

#include
using namespace std;

int num;
vector> edges;
vector indeg;

int main() {

    scanf("%d", &num);
    edges.resize(num);
    indeg.resize(num);
    for(int i=0;i q;
    for(int i=0;i

你可能感兴趣的:(校招,算法,leetcode,c++)