题目
现在你总共有 n 门课需要选,记为 0 到 n-1。
在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们: [0,1]
给定课程总量以及它们的先决条件,返回你为了学完所有课程所安排的学习顺序。
可能会有多个正确的顺序,你只要返回一种就可以了。如果不可能完成所有课程,返回一个空数组。
思路
包含 N个节点的有向图 GG,拓扑排序,判断是否有环,若有环则返回空数组
参考解题:https://leetcode-cn.com/problems/course-schedule-ii/solution/tuo-bu-pai-xu-shen-du-you-xian-bian-li-python-dai-/
class Solution {
public int[] findOrder(int numCourses, int[][] prerequisites) {
//思路:广度优先+贪心
if(numCourses <= 0){
return new int[0];
}
// 1.建立邻接表,存储某个节点能够到达的其他节点集合, 即存储每门课程对应的后续课程
HashSet<Integer>[] adj = new HashSet[numCourses];
for(int i = 0; i < numCourses; i++){
adj[i] = new HashSet<>();
}
// 2.入度数组:通过结点的索引,我们能够得到指向这个结点的结点个数。
int[] inDegree = new int[numCourses];
for(int[] p:prerequisites){//[0, 1]代表从1指向0
//1的后续课程有0,加set
adj[p[1]].add(p[0]);
//0的入度数组++
inDegree[p[0]]++;
}
// 3.新建队列,将所有入度为0的节点先进入队列
Queue<Integer> queue = new LinkedList<>();
for(int i = 0; i < numCourses; i++){
if(inDegree[i] == 0){
queue.offer(i);
}
}
//4. 当队列非空,就从队首取出入度为0的结点,将这个结点输出到结果集中,并且将这个结点的所有邻接结点(它指向的结点)的入度减1,在减1以后,如果这个结点的入度为0,就继续入队。
int[] res = new int[numCourses];
int count = 0;
while(!queue.isEmpty()){
Integer cur = queue.poll();
res[count] = cur;
count++;
Set<Integer> successors = adj[cur];//set指这个结点的所有邻接结点
for(Integer i: successors){
inDegree[i]--;
if(inDegree[i] == 0){
queue.offer(i);
}
}
}
//5. 当队列为空的时候,检查结果集中的顶点个数是否和课程数相等即可。
if(count == numCourses){//不可能完成所有课程,返回一个空数组。
return res;
}
return new int[0];
}
}