LeetCode题库-210题-课程表2

题目描述

程序代码

这个题目就和207题一模一样，只是207题要判断是不是可以全部选完课程，这里要求怎么去选课程，那么我们在最后的队列遍历（BFS）那里记录我们拓扑排序的课程顺序就好了，具体代码如下：

import java.util.*;

public class t210 {
    public int[] findOrder(int numCourses, int[][] prerequisites) {
        int[] ans=new int[numCourses];//标记每个节点的入度
        List<List<Integer>> next=new ArrayList<>();//记录学完一门课程可以继续学那些课程，就是记录课程之间的关系
        Queue<Integer> queue=new LinkedList<>();//记录入度为0的课程，表示最先开始的课程
        //初始化next
        for (int i = 0; i < numCourses; i++) {
            next.add(new ArrayList<>());
        }
        //记录ans和next
        for(int[] temp:prerequisites){
            ans[temp[0]]++;
            next.get(temp[1]).add(temp[0]);
        }
        //初始化queue队列
        for (int i = 0; i < numCourses; i++) {
            if (ans[i]==0) queue.add(i);
        }
        //记录好过程
        int[] ret=new int[numCourses];
        int count=0;
        while(!queue.isEmpty()){
            int temp=queue.poll();
            //唯一改变的地方
            ret[count]=temp;
            count++;

            //学习完这门课程，更新其他课程的入度，如果入度为0，将课程加入queue里面
            for(int res:next.get(temp)){
                if(--ans[res]==0) queue.add(res);
            }
        }
        if(count==numCourses) return ret;
        return new int[0];
    }
}