207. Course Schedule (Graph)

Queue使用时要尽量避免Collection的add()和remove()方法，而是要使用offer()来加入元素，使用poll()来获取并移出元素。它们的优点是通过返回值可以判断成功与否，add()和remove()方法在失败的时候会抛出异常。 如果要使用前端而不移出该元素，使用
element()或者peek()方法。

 

初步了解BFS 用一个queue

题目中用edges的数组表达graph，可以用arraylist[]来组成比较方便使用的graph

 

 

class Solution {
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        ArrayList[] graph = new ArrayList[numCourses];
        int[] degree = new int[numCourses];
        
        for(int i = 0; i < numCourses; i++) {
            degree[i] = 0;
            graph[i] = new ArrayList<>(); //每个arraylist都要new一下
        }
        
        for(int i = 0; i < prerequisites.length; i++) {
            degree[prerequisites[i][0]]++;
            graph[prerequisites[i][1]].add(prerequisites[i][0]);
        }
        int count = 0;
        Queue queue = new LinkedList<>();
        for(int i = 0; i < numCourses; i++) {
            if(degree[i] == 0) {
                count++;
                queue.offer(i);
            }
        }
        while(queue.size() != 0) {
            int a = (int)queue.poll();
            queue.remove(a);
            for(Object i : graph[a]) {
                int b = (int)i;
                degree[b]--;
                if(degree[b] == 0) {
                    count++;
                    queue.offer(b);
                }
                
            }
        }
        return count == numCourses;
        
    }
}