什么人一学期要上2000节课啊jpg
看了非常久都没思路,主要是数据结构还没复习到图论,根本没思路怎么储存一个图……
唯一记得的就是两种存储方法,一种是二维数组法,记录每一条边的有无,一种是只记录有的边,适合于边比较少的情况。
但是两种记录方法在这好像都不太行。
最终还是看了答案。
记录两个东西,一是每一个节点的入度(就是需要上完多少节课才能上当前这节课),一是每个节点出去的线(就是必须上完这节课可以上的其他课)。
遍历整个数组,记录完这两个东西之后,再设置一个队列,记录目前正在上的课。
先将目前能上的课加入队列内(度为0的所有课),然后如果队列不空,就踢出队列的头元素,将头元素出去的线连接的课的出度减一,如果出度为零就加入队列尾部。
这个过程中记录从栈中踢出的课,最终判断课数是否等于课的总数,即位最终结果。
class Solution {
public:bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {vector<vector<int>> connect(numCourses,vector<int>{});int degree[2000];for(int i=0;i<prerequisites.size();i++){connect[prerequisites[i][1]].push_back(prerequisites[i][0]);degree[prerequisites[i][0]]++;}int result=0;queue<int> school;for(int i=0;i<numCourses;i++) if(degree[i]==0) school.push(i);while(!school.empty()){int clas=school.front();school.pop();result++;for(int i=0;i<connect[clas].size();i++){degree[connect[clas][i]]--;if(degree[connect[clas][i]]==0) school.push(connect[clas][i]);}}return result==numCourses;}
};