210. 课程表 II（leetcode210，ArrayList类型的数组创建，拓扑排序）-------------------Java实现

210. 课程表 II（leetcode210，ArrayList类型的数组创建，拓扑排序）-------------------Java实现

题目表述

现在你总共有 numCourses 门课需要选，记为 0 到 numCourses - 1。给你一个数组 prerequisites ，其中 prerequisites[i] = [ai, bi] ，表示在选修课程 ai 前 必须 先选修 bi 。

例如，想要学习课程 0 ，你需要先完成课程 1 ，我们用一个匹配来表示：[0,1] 。
返回你为了学完所有课程所安排的学习顺序。可能会有多个正确的顺序，你只要返回 任意一种 就可以了。如果不可能完成所有课程，返回 一个空数组 。

样例

输入：numCourses = 4, prerequisites = [[1,0],[2,0],[3,1],[3,2]]
输出：[0,2,1,3]
解释：总共有 4 门课程。要学习课程 3，你应该先完成课程 1 和课程 2。并且课程 1 和课程 2 都应该排在课程 0 之后。
因此，一个正确的课程顺序是 [0,1,2,3] 。另一个正确的排序是 [0,2,1,3] 。

条件

1 <= numCourses <= 2000
0 <= prerequisites.length <= numCourses * (numCourses - 1)
prerequisites[i].length == 2
0 <= ai, bi < numCourses
ai != bi
所有[ai, bi] 互不相同

思路

拓扑排序

注意点

注意！ 第一个才是尾结点。

ac代码

Java:

package leetcode210;

import java.util.*;

class Solution {
    public int[] findOrder(int numCourses, int[][] prerequisites) {
        ArrayList<Integer> s_to_target[] = new ArrayList[numCourses];
        for (int i=0;i<s_to_target.length;i++)
            s_to_target[i] = new ArrayList<>();
        int[] pointed = new int[numCourses];
        int[] result = new int[numCourses];
        int sum = 0;
        for (int [] row:prerequisites) {
            s_to_target[row[1]].add(row[0]);
            pointed[row[0]]++;
        }
        Deque<Integer> queue = new LinkedList<>();

        for (int i=0;i<numCourses;i++)
            if (pointed[i]==0)
                queue.offerFirst(i);
        while(!queue.isEmpty()){
            int source = queue.pollFirst();
            result[sum++] = source;
            for (Integer now:s_to_target[source])
            {
                pointed[now]--;
                if (pointed[now]==0)
                    queue.offerFirst(now);
            }
        }
        return (sum==numCourses)?result:new int[0];
    }
}
public class leetcode210 {

    public static void main(String[] args) {
        Scanner cin = new Scanner(System.in);
        int numCourses = cin.nextInt();
        cin.nextLine();
        int sum=0;
        int max = 4000001;
        int[][] prerequisites = new int[max][2];
        for (int i=0;i<max;i++)
        {
            String line = cin.nextLine();
            if (line.isEmpty())
            {
                sum = i;
                break;
            }
            String[] s = line.split(" ");
            prerequisites[i] = new int[s.length];
            for (int j=0;j<s.length;j++)
                prerequisites[i][j] = Integer.parseInt(s[j]);

        }
        int [][] prerequisite = Arrays.copyOfRange(prerequisites,0,sum);
        Solution s = new Solution();
        int[] result = s.findOrder(numCourses,prerequisite);
        System.out.println("result:");
        for (int now :result)
            System.out.print(now+" ");
    }
}
###input:
4
1 0
2 0
3 1
3 2

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。