【数据结构】F:B DS图_课程表 拓扑排序实现

F : B DS图_课程表

Description

小明这个学期必须选修n门课程,课程编号记为0到n-1。

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

例如,先修课程对[0, 1]表示:要想学习课程0,则需要先完成课程1。

请判断小明能否完成所有课程的学习,如果可以则输出true,否则输出false。

Input

第一行输入t,表示有t个测试样例。

接着输入n,表示有n门课程,接着输入len,表示prerequisites数组的长度,接着输入prerequisites数组。

以此类推,共输入t个测试样例。

Output

每一行输出能否完成所有课程的学习,若可以则输出true,否则输出false。

共输出t行。

输入样例:

复制代码
3

2
1
1 0

2
2
1 0
0 1

3
3
1 0
2 0
2 1

输出样例:

复制代码
true
false
true

Hint

复制代码
1 <= n <= 10^5
len == prerequisites.length
1 <= len <= 5000
prerequisites[i].length == 2
0 <= a, b < n
prerequisites中的所有课程对互不相同。

解题思路:

这个问题可以通过拓扑排序来解决。我们需要检查给定的先修课程列表(prerequisites)是否会形成一个有向图中的环。如果有环存在,那么小明就无法完成所有课程(因为有环意味着存在无法满足的先修条件)。拓扑排序可以帮助我们检测这样的环。因为它会把所有入度等于0的节点给挑走,所以它会留下环!如果是一个环的话,即使拓扑排序把其他入度为0的节点给挑走,它的入度也不会是0的,如果完成的课程数量等于课程总数,则返回true,否则返回false

AC代码

cpp 复制代码
#include <iostream>
#include <queue>
using namespace std;

// SZTU forever!!!
// SZTU forever!!!
// SZTU forever!!!

bool canFinish(int numCourses, int** prerequisites, int preLen) {
    int* inDegree = new int[numCourses]();
    int** graph = new int* [numCourses];
    for (int i = 0; i < numCourses; ++i) {
        graph[i] = new int[numCourses]();
    }

    // 建立图并计算入度
    for (int i = 0; i < preLen; ++i) {
        int a = prerequisites[i][0];
        int b = prerequisites[i][1];
        graph[b][a] = 1; // 有向图的边
        inDegree[a]++;
    }

    queue<int> q;
    // 将所有入度为0的课程加入队列
    for (int i = 0; i < numCourses; ++i) {
        if (inDegree[i] == 0) {
            q.push(i);
        }
    }

    // 执行拓扑排序
    int count = 0;
    while (!q.empty()) {
        int current = q.front();
        q.pop();
        count++;
        for (int i = 0; i < numCourses; ++i) {
            if (graph[current][i] == 1 && --inDegree[i] == 0) {
                q.push(i);
            }
        }
    }

    // 清理资源
    for (int i = 0; i < numCourses; ++i) {
        delete[] graph[i];
    }
    delete[] graph;
    delete[] inDegree;

    // 如果计数等于课程数,说明可以完成所有课程
    return count == numCourses;
}

int main() {
    int t;
    cin >> t;
    while (t--) {
        int n, len;
        cin >> n >> len;
        int** prerequisites = new int* [len];
        for (int i = 0; i < len; ++i) {
            prerequisites[i] = new int[2];
            cin >> prerequisites[i][0] >> prerequisites[i][1];
        }

        cout << (canFinish(n, prerequisites, len) ? "true" : "false") << endl;

        for (int i = 0; i < len; ++i) {
            delete[] prerequisites[i];
        }
        delete[] prerequisites;
    }
    return 0;
}
相关推荐
yiridancan18 分钟前
终极剖析HashMap:数据结构、哈希冲突与解决方案全解
java·数据结构·算法·哈希算法
满分观察网友z20 分钟前
性能优化大作战:从 O(N*M) 到 O(N),我的哈希表奇遇记(1865. 找出和为指定值的下标对)
算法
点云SLAM2 小时前
二叉树算法详解和C++代码示例
数据结构·c++·算法·红黑树·二叉树算法
今天背单词了吗9809 小时前
算法学习笔记:19.牛顿迭代法——从原理到实战,涵盖 LeetCode 与考研 408 例题
笔记·学习·算法·牛顿迭代法
没书读了9 小时前
考研复习-数据结构-第六章-图
数据结构
jdlxx_dongfangxing10 小时前
进制转换算法详解及应用
算法
why技术11 小时前
也是出息了,业务代码里面也用上算法了。
java·后端·算法
future141212 小时前
C#进阶学习日记
数据结构·学习
2501_9228955812 小时前
字符函数和字符串函数(下)- 暴力匹配算法
算法
IT信息技术学习圈12 小时前
算法核心知识复习:排序算法对比 + 递归与递推深度解析(根据GESP四级题目总结)
算法·排序算法