【C语言刷LeetCode】207. 课程表(M)

现在你总共有 n 门课需要选,记为 0 到 n-1。

在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们: [0,1]

给定课程总量以及它们的先决条件,判断是否可能完成所有课程的学习?

示例 1:

输入: 2, [[1,0]] 
输出: true
解释: 总共有 2 门课程。学习课程 1 之前,你需要完成课程 0。所以这是可能的。
示例 2:

输入: 2, [[1,0],[0,1]]
输出: false
解释: 总共有 2 门课程。学习课程 1 之前,你需要先完成​课程 0;并且学习课程 0 之前,你还应先完成课程 1。这是不可能的。

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

这道题难。难在工作中几乎没接触过图,所以此题要用到邻接矩阵和入度的知识点时便傻眼了。

然后还需要用数组模拟队列。

先定义二维数组 graph 来表示这个有向图,一维数组inDegree 来表示每个顶点的入度。

我们开始先根据输入来建立这个有向图,邻接矩阵方式,并将入度数组也初始化好。

然后我们定义一个 queue 的数组模拟队列,将所有入度为0的点放入队列中。

接着开始遍历队列,从 graph 里遍历其连接的点,每到达一个新节点,将其入度减一,如果此时该点入度为0,则放入队列末尾。

直到遍历完队列中所有的值,若此时还有节点的入度不为0,则说明环存在,返回 false,反之则返回 true

int queue[10240]= {0};
int head = 0;
int tail = 0;

void queadd(int member) {
    queue[tail] = member;
    tail++;
}

int quepop() {
    int member = 0;
    member = queue[head];
    head++;

    return member;
}

void queinit() {
    memset(queue, 0, sizeof(int) * 10240);
    head = 0;
    tail = 0;
}
    
bool canFinish(int numCourses, int** prerequisites, int prerequisitesSize, int* prerequisitesColSize){
    int graph[numCourses][numCourses];
    int inDegree[numCourses];
    int i;
    int u;
    int num = 0;
    
    memset(graph, 0, sizeof(int) * numCourses * numCourses);
    memset(inDegree, 0, sizeof(int) * numCourses);
    
    queinit();
    
    for(i = 0; i < prerequisitesSize; i++) {
        graph[prerequisites[i][0]][prerequisites[i][1]] = 1;
        inDegree[prerequisites[i][1]]++;
    }
    
    for(i = 0; i < numCourses; i++) {
        if(inDegree[i] == 0){
            queadd(i);
        }
    }
    
    while(head < tail) {
        u = quepop();
        for(i = 0; i < numCourses; i++) {
            if(graph[u][i] != 0) {
                inDegree[i]--;
                if(inDegree[i] == 0){
                    queadd(i);
                }
            }
        }
        num++;
    }
    
    return (num == numCourses);    
}
发布了124 篇原创文章 · 获赞 17 · 访问量 11万+

猜你喜欢

转载自blog.csdn.net/jin615567975/article/details/104382277