拓扑排序-----leetcode 课程表||

现在你总共有 n 门课需要选,记为 0 到 n-1。

在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们: [0,1]

给定课程总量以及它们的先决条件,返回你为了学完所有课程所安排的学习顺序。

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

示例 1:

输入: 2, [[1,0]]
输出: [0,1]
解释: 总共有 2 门课程。要学习课程 1,你需要先完成课程 0。因此,正确的课程顺序为 [0,1] 。
示例 2:

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

这一题就是要我们求出所给数据的一个拓扑排序,我的想法是这样的:
我们先建立一个数组来存放0–n-1这n个数每个课程的入度,然后我们找入度为0 的点,如果找到了标记一下,再将这个点作为起点的那些边的终点的入度-1。

class Solution {
    public int[] findOrder(int numCourses, int[][] prerequisites) {
        int i,j,ans[]=new int[numCourses],in[]=new int[numCourses],count=0;//in 入度
        boolean added[]=new boolean[numCourses],flag=true;
        for(i=0;i<prerequisites.length;i++)
        	in[prerequisites[i][0]]++;
        
        
        while(count<numCourses) {
        	if(!flag) {
        		int fail[]=new int[0];
        		return fail;
        	}
        	flag=false;
        	for(i=0;i<numCourses;i++) {
        		if(!added[i]&&in[i]==0) {//i没有被加入课程表而且i的入度为0
        			ans[count++]=i;
        			added[i]=true;
        			for(j=0;j<prerequisites.length;j++) {
        				if(prerequisites[j][1]==i)
        					in[prerequisites[j][0]]--;
        			}
        			flag=true;
        		}
        	}
        }
        return ans;
    }
}

猜你喜欢

转载自blog.csdn.net/cobracanary/article/details/88682808