210. 课程表 II 【中等题】【拓扑排序】
现在你总共有 n 门课需要选,记为 0 到 n-1。
在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们: [0,1]
给定课程总量以及它们的先决条件,返回你为了学完所有课程所安排的学习顺序。
可能会有多个正确的顺序,你只要返回一种就可以了。如果不可能完成所有课程,返回一个空数组。
输入: 2, [[1,0]]
输出: [0,1]
解释: 总共有 2 门课程。要学习课程 1,你需要先完成课程 0。因此,正确的课程顺序为 [0,1]
输入: 4, [[1,0],[2,0],[3,1],[3,2]]
输出: [0,1,2,3] or [0,2,1,3]
解释: 总共有 4 门课程。要学习课程 3,你应该先完成课程 1 和课程 2。并且课程 1 和课程 2 都应该排在课程 0 之后。因此,一个正确的课程顺序是 [0,1,2,3] 。另一个正确的排序是 [0,2,1,3]
题目讲解
【历史重难点题目】
【核心思想】
- 不断删除入度为0的结点,直到无法删除为止
【数据结构】
- 数组,链表,队列
【思路】
- 将每个课看成一个结点,将各个结点的入度存在数组里
- 将各个结点之间的关系,以邻接链表的形式存储下来
- 不断删除入度为0的结点,同时加入ans的数组中,直到无法删除为止
- 判断最后是否还剩下结点,若还剩下,则有环,不能完成所有课程的学习;若没有剩下,则无环,能完成所有课程的学习,返回ans
【代码】
public int[] findOrder(int numCourses, int[][] prerequisites) {
int[] indegree=new int[numCourses];
List[] list=new ArrayList[numCourses];
int[] ans=new int[numCourses];
for(int i=0;i<prerequisites.length;i++){
indegree[prerequisites[i][0]]++;
if(list[prerequisites[i][1]]==null) {
List<Integer> temp=new ArrayList<>();
temp.add(prerequisites[i][0]);
list[prerequisites[i][1]]=temp;
}
else{
List temp=list[prerequisites[i][1]];
temp.add(prerequisites[i][0]);
list[prerequisites[i][1]]=temp;
}
}
int num=0;
Queue<Integer> queue=new ArrayDeque<>();
for(int i=0;i<numCourses;i++){
if(indegree[i]==0)
queue.add(i);
}
while(!queue.isEmpty()){
int index=queue.poll();
ans[num]=index;
num++;
while(list[index]!=null && !list[index].isEmpty()){
if(--indegree[(int)list[index].get(0)]==0)
queue.offer((int)list[index].get(0));
list[index].remove(0);
}
}
if(num!=numCourses)
return new int[0];
return ans;
}
关注微信公众号“算法岗从零到无穷”,更多算法知识点告诉你。