java算法课程_Java 算法-安排课程(广搜)

其实，在学习以来知道什么是广搜，也知道广搜实现的基本过程，但是从来没有用过，所以有必要认真的理解一下广搜。今天看到了一道广搜的题，尽管知道用广搜的，并且能够成功的推导出来，但是就是不能把代码写出来。

1.概览

(1).题意

你需要去上n门九章的课才能获得offer，这些课被标号为 0 到 n-1 。

有一些课程需要“前置课程”，比如如果你要上课程0，你需要先学课程1，我们

用一个匹配来表示他们： [0,1]

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

习顺序。

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

程，返回一个空数组。

(2).样例

给定 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]

2.解题思路

这个题的基本思路就是将当前能够学习的课全部入队，然后在从队列中取出一个来，表示已经学习，然后再将能够学习的课程全部入队依次这样，知道队列为空。如图所示

所以这里，我们还需要两个辅助的东西，一个是这门的先决课程有几门，一个是这门修完了可以修那些课程。

一门课程额后续课程：

for (int i = 0; i < prerequisites.length; i++) {

list.get(prerequisites[i][1]).add(prerequisites[i][0]);

}

一个课程的先决课程的门数：

for (int i = 0; i < numCourses; i++) {

for (int num : list.get(i)) {

a[num]++;

}

}

3.代码

public int[] findOrder(int numCourses, int[][] prerequisites) {

int a[] = new int[numCourses];

List> list = new ArrayList<>();

for (int i = 0; i < numCourses; i++) {

list.add(new HashSet());

}

for (int i = 0; i < prerequisites.length; i++) {

list.get(prerequisites[i][1]).add(prerequisites[i][0]);

}

for (int i = 0; i < numCourses; i++) {

for (int num : list.get(i)) {

a[num]++;

}

}

Queue queue = new LinkedList<>();

for (int i = 0; i < numCourses; i++) {

if (a[i] == 0) {

queue.offer(i);

}

}

int courses[] = new int[numCourses];

int count = 0;

while (!queue.isEmpty()) {

int num = queue.poll();

for (int n : list.get(num)) {

a[n]--;

if (a[n] == 0) {

queue.offer(n);

}

}

courses[count++] = num;

}

if (count == numCourses) {

Arrays.stream(courses).forEach(System.out::print);

}

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

}