题目
你这个学期必须选修 numCourse 门课程,记为 0 到 numCourse-1 。
在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们:[0,1]
给定课程总量以及它们的先决条件,请你判断是否可能完成所有课程的学习?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/course-schedule
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
Java
思路
①统计课程安排图中每个节点的入度,生成入度表 rd;
②借助一个队列 queue,将所有入度为0的节点入队;
③当 queue 非空时,依次将队首节点出队,在课程安排图中删除此节点;
④若整个课程安排图是有向无环图,则所有节点一定都入队并出队过,即完成拓扑排序。换个角度说,若课程安排图中存在环,一定有节点的入度始终不为 0。
因此,拓扑排序出队次数等于课程个数,返回 count==numCourses?true:false; 判断课程是否可以成功安排。
public class Solution {
public boolean canFinish(int numCourses, int[][] prerequisites) {
int[][] Group=new int[numCourses][numCourses];
int[] rd=new int[numCourses];//记录入度的数组
for(int i=0;i<prerequisites.length;i++)
{
Group[prerequisites[i][1]][prerequisites[i][0]]=1;//prerequisites[i][1]到prerequisites[i][0]存在通路
rd[prerequisites[i][0]]++;//入度加一
}
Queue<Integer> queue=new LinkedList<>();//定义一个队列用来存储先修课学完的课程
for(int j=0;j<numCourses;j++)
{
if(rd[j]==0)//先修课已学完
{
queue.add(j);//入队
}
}
int count=0;
while(!queue.isEmpty())//队列非空,还有课程待学
{
int temp=queue.poll();//记录此时要学的课程
for(int k=0;k<numCourses;k++)
{
if(Group[temp][k]!=0)//找到Group中temp对应的边中未学的课程
{
rd[k]--;//入度-1,即temp这个课程已学
if(rd[k]==0)//入度为0,入队
{
queue.add(k);
}
}
}
count++;//记录学了的课程
}
return count==numCourses?true:false;//学完的课程等于总的课程即完成任务,反之则反
}
}