题目描述(困难)
这里有 n 门不同的在线课程,按从 1 到 n 编号。给你一个数组 courses ,其中 courses[i] = [durationi, lastDayi] 表示第 i 门课将会 持续 上 durationi 天课,并且必须在不晚于 lastDayi 的时候完成。
你的学期从第 1 天开始。且不能同时修读两门及两门以上的课程。
返回你最多可以修读的课程数目。
示例 1:
输入:courses = [[100, 200], [200, 1300], [1000, 1250], [2000, 3200]]
输出:3
解释:
这里一共有 4 门课程,但是你最多可以修 3 门:
首先,修第 1 门课,耗费 100 天,在第 100 天完成,在第 101 天开始下门课。
第二,修第 3 门课,耗费 1000 天,在第 1100 天完成,在第 1101 天开始下门课程。
第三,修第 2 门课,耗时 200 天,在第 1300 天完成。
第 4 门课现在不能修,因为将会在第 3300 天完成它,这已经超出了关闭日期。
示例 2:
输入:courses = [[1,2]]
输出:1
示例 3:
输入:courses = [[3,2],[4,3]]
输出:0
提示:
1 <= courses.length <= 10^4
1 <= durationi, lastDayi <= 10^4
思路1,贪心+回溯(超时)
对课程按要求结束时间排序,然后就是选或不选的问题,用回溯来做,思路对但超时
代码
//超时
class Solution {
public:
vector<int> ans;
static bool cmp(vector<int>& x, vector<int>& y) {
if(x[1] < y[1]) return true;
return false;
}
void dfs(vector<vector<int>>& courses, int time, int count, int cno) {
if(cno == courses.size()) {
ans.push_back(count);
return ;
}
if(time + courses[cno][0] <= courses[cno][1]) {
dfs(courses, time + courses[cno][0], count+1, cno+1);
}
dfs(courses, time, count, cno+1);
}
int scheduleCourse(vector<vector<int>>& courses) {
sort(courses.begin(), courses.end(), cmp);
dfs(courses, 0, 0, 0);
int res = 0;
for(auto c : ans) {
res = max(res, c);
}
return res;
}
};
思路2,贪心+优先队列
考虑一种反悔贪心策略
如果遇到不能选课了,我把过往选过的课程中,去掉耗时最长的(性价比最低)的课程,这样我就有更多的空闲时间去尽可能多的课程,而记录课程耗时并方便找出最大值的数据结构就是优先队列。
所以每次如果能选课,则加入优先队列,不能选课,就弹出耗时最大的再来判断,最后优先队列中的课程数就是最大可选课程数。
代码
class Solution {
public:
static bool cmp(vector<int>& x, vector<int>& y) {
if(x[1] < y[1]) return true;
return false;
}
int scheduleCourse(vector<vector<int>>& courses) {
sort(courses.begin(), courses.end(), cmp);
priority_queue<int> q;
// 优先队列中所有课程的总时间
int total = 0;
for (const auto& course: courses) {
int ti = course[0], di = course[1];
if (total + ti <= di) {
total += ti;
q.push(ti);
}
else if (!q.empty() && q.top() > ti) {
total -= q.top() - ti;
q.pop();
q.push(ti);
}
}
return q.size();
}
};
另外补充个排序更简洁的写法
sort(courses.begin(), courses.end(), [](const auto& c0, const auto& c1) {
return c0[1] < c1[1];
});