来源:力扣(LeetCode)
描述:
给你一个整数 n ,表示有 n 节课,课程编号从 1 到 n 。同时给你一个二维整数数组 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
- 先修课程图是一个有向无环图。
方法:记忆化搜索
思路
要求出完成所有课程的最少月份数,可以求出每门课程的最少月份数,然后求出最大值。首先根据 relations,构建先修课临接表 prev,prev[i] 就表示课程 i 的所有的先修课。定义函数 dp,输入参数为 i,返回完成课程 i 所需的最少月份数。
- 如果一门课程 i 没有先修课要求,那么完成它的最少月份数就是 time[i−1]。
- 如果一门课有先修课时,完成它的最少月份数就是在它的所有先修课的最少完成月份的最大值的基础上,再加上 time[i−1],即 dp[i] = max(dp[j]) + time[i−1] ,j ∈ prev[i]。
可以运用记忆化搜索的技巧,求出每门课的最少完成月份数。因为运用了记忆化搜索,每门课的最少完成月份数最多只会被计算一次。
代码:
class Solution {
public:
int minimumTime(int n, vector<vector<int>>& relations, vector<int>& time) {
int mx = 0;
vector<vector<int>> prev(n + 1);
for (auto &relation : relations) {
int x = relation[0], y = relation[1];
prev[y].emplace_back(x);
}
unordered_map<int, int> memo;
function<int(int)> dp = [&](int i) -> int {
if (!memo.count(i)) {
int cur = 0;
for (int p : prev[i]) {
cur = max(cur, dp(p));
}
cur += time[i - 1];
memo[i] = cur;
}
return memo[i];
};
for (int i = 1; i <= n; i++) {
mx = max(mx, dp(i));
}
return mx;
}
};
执行用时:384 ms, 在所有 C++ 提交中击败了53.19%的用户
内存消耗:161.1 MB, 在所有 C++ 提交中击败了5.32%的用户
复杂度分析
时间复杂度:O(m+n),其中 m 是数组 relations 长度。需要构建先修课临接表,并且计算每个课程的最少月份数。因为每个课程只会被计算一次,因此相当于是每个 relation 会被遍历一次。
空间复杂度:O(m+n),先修课临接表的空间复杂度是 O(m+n),记忆化搜索的空间复杂度是 O(n)。
author:LeetCode-Solution