按排课程

你需要去上n门九章的课才能获得offer，这些课被标号为 0 到 n-1 。
有一些课程需要“前置课程”，比如如果你要上课程0，你需要先学课程1，我们用一个匹配来表示他们： [0,1]

给你课程的总数量和一些前置课程的需求，返回你为了学完所有课程所安排的学习顺序。

可能会有多个正确的顺序，你只要返回一种就可以了。如果不可能完成所有课程，返回一个空数组。

样例

给定 n = 2, prerequisites = [[1,0]]
返回 [0,1]

给定 n = 4, prerequisites = [1,0],[2,0],[3,1],[3,2]]
返回 [0,1,2,3] or [0,2,1,3]

思路 ：

拓扑排序 广度优先

代码:

 

import java.util.*;


public class Solution
{
    public int[] findOrder(int n, int[][] pre)
    {
int[] preNum = new int[n];//记录入度的数量
HashMap<Integer, ArrayList<Integer>> bacItem = new HashMap<Integer, ArrayList<Integer>>();//记录后置元素

//初始化
for (int i = 0; i < n; i++)
{
    ArrayList<Integer> tmp = new ArrayList<Integer>();
    bacItem.put(i, tmp);
}


for (int i = 0; i < pre.length; i++)
{
    int start = pre[i][1];
    int end = pre[i][0];
    preNum[end]++;
    bacItem.get(start).add(end);
}

Deque<Integer> que = new LinkedList<Integer>();
List<Integer> ans = new ArrayList<Integer>();
for (int i = 0; i < preNum.length; i++)
{
    if (preNum[i] == 0)//把所有入度为0的点入队列
    {
que.addFirst(i);
    }
}

//广度优先
while (!que.isEmpty())
{
    int t = que.pollLast();
    ans.add(t);
    ArrayList<Integer> list = bacItem.get(t);
    for (Integer e : list)
    {
preNum[e]--;
if (preNum[e] == 0)
{
    que.add(e);
}
    }
}


if (ans.size() == n)
{
    int[] tmp = new int[n];
    for (int i = 0; i < n; i++)
tmp[i] = ans.get(i);
    return tmp;
} else
    return new int[] {};
    }
}