算法:拓扑排序-课程表

class Solution {

    public boolean canFinish(int numCourses, int[][] prerequisites) {
        List<Integer>[] arr = new ArrayList[numCourses];
        int[] inDegree = new int[numCourses];
        for(int i=0; i<numCourses; i++) {
            arr[i] = new ArrayList<Integer>();
        }
        for(int i=0; i<prerequisites.length; i++) {
            arr[prerequisites[i][0]].add(prerequisites[i][1]);
            inDegree[prerequisites[i][1]]++;
        }
        Stack<Integer> stack = new Stack<>();
        for(int i=0; i<numCourses; i++) {
            if(inDegree[i] == 0) {
                stack.push(i);
            }
        }
        int cnt = 0;
        while(!stack.isEmpty()) {
            int del = stack.pop();
            cnt++;
            for(int i=0; i<arr[del].size(); i++) {
                int tmp = arr[del].get(i);
                inDegree[tmp]--;
                if(inDegree[tmp] == 0) {
                    stack.push(tmp);
                }
            }
        }
        return cnt == numCourses;
    }

}
全部评论

相关推荐

勤奋努力的椰子这就开摆:美团骑手在美团工作没毛病
投递美团等公司10个岗位
点赞 评论 收藏
分享
10-30 23:23
已编辑
中山大学 Web前端
去B座二楼砸水泥地:这无论是个人素质还是专业素质都👇拉满了吧
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务