什么人一学期要上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;
}
};