【C++BFS】1462. 课程表 IV

最新推荐文章于 2024-11-13 19:36:04 发布

闻缺陷则喜何志丹

最新推荐文章于 2024-11-13 19:36:04 发布

阅读量999

点赞数 28

文章标签： c++ 宽度优先力扣算法课程表先决条件查询

本文链接：https://blog.csdn.net/he_zhidan/article/details/140587546

版权

工作级难度算法题解专栏收录该内容

195 篇文章 1 订阅

订阅专栏

本文涉及知识点

C++BFS算法

LeetCode1462. 课程表 IV

你总共需要上 numCourses 门课，课程编号依次为 0 到 numCourses-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 <= 10⁴
0 <= ui, vi <= n - 1
ui != vi

大致步骤

一，通过BFS枚举各节点的后续节点。vis[i][j]表示i是否是j的前置节点。
二，依次处理各出现。
总时间复杂度：O(n³)

C++BFS

BFS的预处理：生成邻接表neiBo。
BFS的状态表示：leves[0] = {root} leves[i]记录root 的i级后续节点。空间复杂度:O(n)
BFS的状态表示：通过next枚举cur的后续节点。时间复杂度：O(nn)
BFS的初始状态：leves[0] = {root}
BFS的返回值：无
BFS的出重：vis[i]

代码

核心代码

class Solution {
		public:
			vector<bool> checkIfPrerequisite(int N, vector<vector<int>>& prerequisites, vector<vector<int>>& queries) {
				vector<vector<int>> neiBo(N);
				for (const auto& v : prerequisites) {
					neiBo[v[0]].emplace_back(v[1]);
				}
				vector<vector<bool>> vis(N, vector<bool>(N));
				auto BFS = [&](int root, vector<bool>& vis) {
					vis[root] = true;
					queue<int> que;
					que.emplace(root);
					while (que.size()) {
						const auto cur = que.front();
						que.pop();
						for (const auto& next : neiBo[cur]) {
							if (vis[next]) { continue; }
							vis[next] = true;
							que.emplace(next);
						}
					}
				};
				for (int i = 0; i < N; i++) {
					BFS(i, vis[i]);
				}
				vector<bool> ret;
				for (const auto& v : queries) {
					ret.emplace_back(vis[v[0]][v[1]]);
				}
				return ret;
			}
		};

单元测试

int numCourses;
		vector<vector<int>> prerequisites, queries;
		TEST_METHOD(TestMethod11)
		{
			numCourses = 2, prerequisites = { {1,0} }, queries = { {0,1},{1,0} };
			auto res = Solution().checkIfPrerequisite(numCourses, prerequisites, queries);
			AssertEx(vector<bool>{false,true}, res);
		}
		TEST_METHOD(TestMethod12)
		{
			numCourses = 2, prerequisites = {}, queries = { {1,0},{0,1} };
			auto res = Solution().checkIfPrerequisite(numCourses, prerequisites, queries);
			AssertEx(vector<bool>{false, false}, res);
		}
		TEST_METHOD(TestMethod13)
		{
			numCourses = 3, prerequisites = { {1,2},{1,0},{2,0} }, queries = { {1,0},{1,2} };
			auto res = Solution().checkIfPrerequisite(numCourses, prerequisites, queries);
			AssertEx(vector<bool>{true, true}, res);
		}
		TEST_METHOD(TestMethod14)
		{
			numCourses = 5, prerequisites = { {4,3},{4,1},{4,0},{3,2},{3,1},{3,0},{2,1},{2,0},{1,0} },
				queries = { {1,4},{4,2},{0,1},{4,0},{0,2},{1,3},{0,1} };
			auto res = Solution().checkIfPrerequisite(numCourses, prerequisites, queries);
			AssertEx(vector<bool>{false, true, false, true, false, false, false}, res);
		}

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功