力扣:2050. 并行课程 III

给你一个整数 n ,表示有 n 节课,课程编号从 1n 。同时给你一个二维整数数组 relations ,其中 relations[j] = [prevCoursej, nextCoursej] ,表示课程 prevCoursej 必须在课程 nextCoursej 之前 完成(先修课的关系)。同时给你一个下标从 0 开始的整数数组 time ,其中 time[i] 表示完成第 (i+1) 门课程需要花费的 月份 数。

请你根据以下规则算出完成所有课程所需要的 最少 月份数:

  • 如果一门课的所有先修课都已经完成,你可以在 任意 时间开始这门课程。
  • 你可以 同时任意门课程

请你返回完成所有课程所需要的 最少 月份数。

**注意:**测试数据保证一定可以完成所有课程(也就是先修课的关系构成一个有向无环图)。

示例 1:

复制代码
输入:n = 3, relations = [[1,3],[2,3]], time = [3,2,5]
输出:8
解释:上图展示了输入数据所表示的先修关系图,以及完成每门课程需要花费的时间。
你可以在月份 0 同时开始课程 1 和 2 。
课程 1 花费 3 个月,课程 2 花费 2 个月。
所以,最早开始课程 3 的时间是月份 3 ,完成所有课程所需时间为 3 + 5 = 8 个月。

示例 2:

复制代码
输入:n = 5, relations = [[1,5],[2,5],[3,5],[3,4],[4,5]], time = [1,2,3,4,5]
输出:12
解释:上图展示了输入数据所表示的先修关系图,以及完成每门课程需要花费的时间。
你可以在月份 0 同时开始课程 1 ,2 和 3 。
在月份 1,2 和 3 分别完成这三门课程。
课程 4 需在课程 3 之后开始,也就是 3 个月后。课程 4 在 3 + 4 = 7 月完成。
课程 5 需在课程 1,2,3 和 4 之后开始,也就是在 max(1,2,3,7) = 7 月开始。
所以完成所有课程所需的最少时间为 7 + 5 = 12 个月。

提示:

  • 1 <= n <= 5 * 104
  • 0 <= relations.length <= min(n * (n - 1) / 2, 5 * 104)
  • relations[j].length == 2
  • 1 <= prevCoursej, nextCoursej <= n
  • prevCoursej != nextCoursej
  • 所有的先修课程对 [prevCoursej, nextCoursej] 都是 互不相同 的。
  • time.length == n
  • 1 <= time[i] <= 104
  • 先修课程图是一个有向无环图。

思路:

首先初始化入度数组indegree统计每门课程(结点)的前置依赖数量,通过大小为5e4 + 10的邻接表a构建课程依赖有向图,同时定义dist数组记录完成每门课程的最早时间;

遍历先修关系数组relations,将[u, v](修 v 需先修 u)转化为从 u 指向 v 的有向边,更新 v 的入度并填充邻接表。

接着筛选出所有入度为 0 的课程,若该课程无后续课程则直接将其学习时长存入数组b,否则加入队列作为排序起点;

随后循环取出队首课程u,若u无后续课程则将其完成时间(dist[u] + time[u-1])存入b,否则遍历其所有后续课程v,先将v的入度减 1,再更新v的最早完成时间为 "当前值" 与 "u的完成时间 + u的学习时长" 的较大值(保证v在所有先修课完成后开始),若v入度变为 0 则加入队列。

最后对存储所有课程完成时间的数组b排序,取最大值作为完成所有课程的最短时间。

AC代码:

cpp 复制代码
class Solution {
public:
	int minimumTime(int n, vector<vector<int>>& relations, vector<int>& time) {
		vector<int> indegree(n + 1, 0); // 计算入度
		const int N = 5e4 + 10;
		vector<int> a[N]; // 邻接表建图
		vector<int> dist(n + 1, 0);
		for (auto it : relations) {
			int u = it[0];
			int v = it[1];
			indegree[v]++;
			a[u].push_back(v);
		}
		queue<int> q;
		vector<int> b;
		for (int i = 1; i <= n; i++) {
			if (indegree[i] == 0) {
				if (a[i].empty()) {
					b.push_back(time[i - 1]);
				}
				else {
					q.push(i);
				}
				
			}
		}
		while (!q.empty()) {
			int u = q.front();
			q.pop();
			if (a[u].empty()) {
				b.push_back(dist[u] + time[u - 1]);
				continue;
			}

			for (auto v : a[u]) {
				indegree[v]--;
				if (dist[v] < dist[u] + time[u - 1]) {
					dist[v] = dist[u] + time[u - 1];
				}
				if (indegree[v] == 0) {
					q.push(v);
				}
			}
		}
		sort(b.begin(), b.end());
		int ans = b[b.size() - 1];
		return ans;
	}
};
相关推荐
monster000w5 小时前
大模型微调过程
人工智能·深度学习·算法·计算机视觉·信息与通信
小小晓.5 小时前
Pinely Round 4 (Div. 1 + Div. 2)
c++·算法
SHOJYS5 小时前
学习离线处理 [CSP-J 2022 山东] 部署
数据结构·c++·学习·算法
biter down5 小时前
c++:两种建堆方式的时间复杂度深度解析
算法
zhishidi6 小时前
推荐算法优缺点及通俗解读
算法·机器学习·推荐算法
WineMonk6 小时前
WPF 力导引算法实现图布局
算法·wpf
2401_837088506 小时前
双端队列(Deque)
算法
ada7_6 小时前
LeetCode(python)108.将有序数组转换为二叉搜索树
数据结构·python·算法·leetcode
奥特曼_ it6 小时前
【机器学习】python旅游数据分析可视化协同过滤算法推荐系统(完整系统源码+数据库+开发笔记+详细部署教程)✅
python·算法·机器学习·数据分析·django·毕业设计·旅游
仰泳的熊猫6 小时前
1084 Broken Keyboard
数据结构·c++·算法·pat考试