210. 课程表 II

https://leetcode-cn.com/problems/course-schedule-ii/

唉一开始没想到怎么做,就暴力解了,结果到最后几个地方冒出个2000门课直接把我TLE了,我人都傻了你特么大学上2000门课?

马上就改了思路,想了下大三上算法设计与分析的时候老师讲的拓扑排序算法,得出了以下的代码,成绩还算可以~

执行用时:5 ms
内存消耗:41 MB
class Solution {
    public int[] findOrder(int numCourses, int[][] prerequisites) {
         /**
         * 我们在算法中学习的拓扑排序一般是这样讲的:
         * 1.找入度为0的这个节点
         * 2.去除该结点,并将该节点可到达的节点的入度-1
         * 3.寻找下一个入度为0的节点。
         */

        //存放该节点可到达的节点
        List<Integer>[] list = new LinkedList[numCourses];
        for(int i = 0; i < list.length; i++){
            list[i] = new LinkedList<>();
        }
        //统计每个节点的入度
        int[] degree = new int[numCourses];
        //初始化邻接矩阵和入度
        for (int[] prerequisite : prerequisites) {
            degree[prerequisite[0]]++;
            list[prerequisite[1]].add(prerequisite[0]);
        }
        //然后就开始主过程
        List<Integer> temp = new ArrayList<>();
        while(temp.size() != numCourses){
            int i = 0;
            for(; i < numCourses; i++){
                if(degree[i] == 0){
                    break;
                }
            }
            //如果找到越界都找不到,说明存在环,提前结束
            if(i == numCourses){
                break;
            }
            //将该节点入度置-1,表示已经访问过了
            degree[i] = -1;
            //加入到答案list中
            temp.add(i);
            //将该节点能到达的节点的度数减一
            while (!list[i].isEmpty()){
                degree[list[i].remove(0)]--;
            }

        }
        //这个情况就是存在环,直接返回空数组
        if(temp.size() < numCourses){
            return new int[0];
        }
        //将list中的数据导入数组中
        int[] res = new int[numCourses];

        for (int i = 0; i < temp.size(); i++){
            res[i] = temp.get(i);
        }

        return res;
    }
}

需要解释的都在注释里面了,不会有人看不懂吧 不会吧不会吧

posted @ 2020-05-17 13:30  ZJPang  阅读(139)  评论(0编辑  收藏  举报