你需要去上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[] {};
}
}