题目
top排序问题。
选课,学A课,需要先学B课。然后给出一个可能的顺序。
详细见,LeetCode(https://leetcode.cn/problems/course-schedule-ii/)。
题解
思路
top排序
- 选出先导课为0的课程,然后学。
- 因为已经学了,所以这些课程不再是先导课,把它从其它课程的先导课列表中删除,如果列表为空,则将这个课程加入到先导课为0的课程的队伍中。
- 有些课程存在循环,这个时候怎么判断有没有循环呢?就是看最后计算的结果的大小,是否等于课程数即可。
时间复杂度
O(n + m)
空间复杂度
O(n + m)
代码
/*
top排序,建图,每次处理出(或入,取决于您怎么建图)度为0的点,然后更新关联的点
*/
class Solution {
public:
vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
vector<vector<int>> edge(numCourses);
vector<int> degree(numCourses);
for(auto prerequisite : prerequisites) {
edge[prerequisite[1]].push_back(prerequisite[0]);
degree[prerequisite[0]]++;
}
queue<int> q;
for(int i = 0; i < numCourses; i++) {
if(degree[i] == 0) {
q.push(i);
}
}
vector<int> ans;
while(q.size()) {
int top = q.front();
q.pop();
ans.push_back(top);
for(auto next : edge[top]) {
degree[next]--;
if(degree[next] == 0) {
q.push(next);
}
}
}
// 如果存在环,则不满足题意,则返回空
if(ans.size() != numCourses) ans.clear();
return ans;
}
};