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());

    }

}