lintcode安排课程答案

package study.code.lint.answer;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * 你需要去上n门九章的课才能获得offer，这些课被标号为 0 到 n-1 。
 *    有一些课程需要“前置课程”，比如如果你要上课程0，你需要先学课程1，我们用一个匹配来表示他们： [0,1]
 *    给你课程的总数量和一些前置课程的需求，返回你为了学完所有课程所安排的学习顺序。
 *    可能会有多个正确的顺序，你只要返回一种就可以了。如果不可能完成所有课程，返回一个空数组。
 */
public class Solution {
     public static int[] findOrder(int numCourses, int[][] prerequisites) {
         if(numCourses<1||prerequisites==null){
             return new int[0];
         }
         List<HashSet<Integer>> list = new ArrayList<HashSet<Integer>>();
         for(int i=0;i<numCourses;i++){
             list.add(new HashSet<Integer>());
         }
         for(int i=0;i<prerequisites.length;i++){
             list.get(prerequisites[i][1]).add(prerequisites[i][0]); 
         }

         int[] premun = new int[numCourses] ;
         for(int i=0;i<numCourses;i++){
             for(int num :list.get(i)){
                 premun[num]++;
             }
         }

         Queue<Integer> queue = new LinkedList<Integer>();
         for(int i=0;i<numCourses;i++){
             if(premun[i]==0){
                 queue.offer(i);
             }
         }
         int [] result = new int[numCourses] ;
         int count=0;
         while(!queue.isEmpty()){
             int current = queue.poll();
             for(int num :list.get(current)){
                 premun[num]--;
                 if(premun[num]==0){
                     queue.offer(num);
                 }
             }
             result[count++]=current;
         }

         return count ==numCourses?result:new int[0];
     }

     public static void main(String[] args) { 
         int numCourses  = 4;  
         int[][] prerequisites = {{1,0},{0,2},{3,1},{2,3}};
         int [] result = findOrder(numCourses, prerequisites);
         System.out.println(result.toString());
    }
}