LeetCode每日一题:1462. 课程表 IV(2023.9.12 C++)

目录

[1462. 课程表 IV](#1462. 课程表 IV)

题目描述:

实现代码与解析:

拓扑排序

原理思路:


1462. 课程表 IV

题目描述:

你总共需要上 numCourses 门课,课程编号依次为 0numCourses-1 。你会得到一个数组 prerequisite ,其中 prerequisites[i] = [ai, bi] 表示如果你想选 bi 课程,你必须 先选 ai 课程。

  • 有的课会有直接的先修课程,比如如果想上课程 1 ,你必须先上课程 0 ,那么会以 [0,1] 数对的形式给出先修课程数对。

先决条件也可以是 间接 的。如果课程 a 是课程 b 的先决条件,课程 b 是课程 c 的先决条件,那么课程 a 就是课程 c 的先决条件。

你也得到一个数组 queries ,其中 queries[j] = [uj, vj]。对于第 j 个查询,您应该回答课程 uj 是否是课程 vj 的先决条件。

返回一个布尔数组 answer ,其中 answer[j] 是第 j 个查询的答案。

示例 1:

复制代码
输入:numCourses = 2, prerequisites = [[1,0]], queries = [[0,1],[1,0]]
输出:[false,true]
解释:课程 0 不是课程 1 的先修课程,但课程 1 是课程 0 的先修课程。

示例 2:

复制代码
输入:numCourses = 2, prerequisites = [], queries = [[1,0],[0,1]]
输出:[false,false]
解释:没有先修课程对,所以每门课程之间是独立的。

示例 3:

复制代码
输入:numCourses = 3, prerequisites = [[1,2],[1,0],[2,0]], queries = [[1,0],[1,2]]
输出:[true,true]

提示:

  • 2 <= numCourses <= 100
  • 0 <= prerequisites.length <= (numCourses * (numCourses - 1) / 2)
  • prerequisites[i].length == 2
  • 0 <= ai, bi <= n - 1
  • ai != bi
  • 每一对 [ai, bi]不同
  • 先修课程图中没有环。
  • 1 <= queries.length <= 104
  • 0 <= ui, vi <= n - 1
  • ui != vi

实现代码与解析:

拓扑排序

cpp 复制代码
class Solution {
public:
    // 邻接表
    vector<int> e = vector<int>(5010, 0), ne = vector<int>(5010, 0), h = vector<int>(110, -1);
    int idx = 0;

    // 加边
    void add (int a, int b)
    {
        e[idx] = b, ne[idx] = h[a], h[a] = idx++;
    }

    vector<bool> checkIfPrerequisite(int numCourses, vector<vector<int>>& prerequisites, vector<vector<int>>& queries) {
        vector<vector<bool>> sumres(numCourses, vector<bool>(numCourses));
        vector<bool> res;
        vector<int> indeg(numCourses, 0);

        for (auto t: prerequisites)
        {
            add(t[0], t[1]);
            indeg[t[1]]++; // 入度++
        }

        queue<int> q;
        for (int i = 0; i < numCourses; i++)
        {
            if (indeg[i] == 0) q.push(i); // 度为 0 入队
        }

        // bfs
        while (q.size())
        {
            int t = q.front();
            q.pop();

            for (int i = h[t]; ~i; i = ne[i])
            {
                int j = e[i];
                sumres[t][j] = true; // 直接相连true

                indeg[j]--; // 入度--
                if (indeg[j] == 0) q.push(j);

                for (int k = 0; k < numCourses; k++)
                    sumres[k][j] = sumres[k][t] || sumres[k][j]; // 传递, 间接相连

            }
        }

        for (auto t: queries)
        {
            res.push_back(sumres[t[0]][t[1]]);
        }
        return res;
    }
};

原理思路:

bfs拓扑排序。只深搜会超时。

拓扑排序详解(带有C++模板)_Cosmoshhhyyy的博客-CSDN博客

因为课程间接传递也可以,注意把上一个结点信息,传递给下一个结点即可。和最短路算法的最短路更新相似。

相关推荐
罗西的思考13 小时前
机器人 / 强化学习】HIL-SERL:人类在环驱动的具身智能进化框架
人工智能·算法·机器学习
美团技术团队16 小时前
LongCat 开源 VitaBench 2.0:长期动态智能体基准新标杆
人工智能·算法
To_OC1 天前
LC 207 课程表:刚学图论那会儿,我连这是拓扑排序都没看出来
javascript·算法·leetcode
To_OC1 天前
LC 208 实现 Trie 前缀树:曾被名字劝退,写完发现是送分题
javascript·算法·leetcode
BadBadBad__AK1 天前
线段树维护区间 k 次方和
c++·数学·算法·stl
_清歌2 天前
DSpark 深度解读:DeepSeek-V4 如何用「半自回归」把推理速度提升 85%
算法
统计实现局2 天前
SVD 的三步走:双对角化、Givens 收敛、排序
算法
躬行见万象2 天前
《VLA 系列》UniLab 强化训练 | G1 机器人 |复现
算法
统计实现局2 天前
对称不定分解(Bunch-Kaufman):为什么 Cholesky 不够用
算法