原题链接:课程表
个人解法
思路:
一般这种完成某一件事需要完成另一件事,就是典型的AOV网,那么如果能完成所有课程,这个图必须是有向无环图,那么我们可以利用拓扑排序的思路,求图的拓扑序,如果存在环,则无法遍历到所有的结点,可以用一个计数量来表示。
时间复杂度: O ( n + m ) O(n+m) O(n+m)
代码:
class Solution {
public:
int h[100010];
int e[200010], ne[200010], idx;
int d[100010];
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
bool topsort(int &numCourses) {
int cnt = 0;
queue<int> q;
for(int i = 0;i < numCourses;i ++) {
if(!d[i]) q.push(i);
}
while(q.size()) {
int t = q.front();
q.pop();
cnt ++;
for(int i = h[t];i != -1;i = ne[i]) {
int j = e[i];
if(-- d[j] == 0) q.push(j);
}
}
return numCourses == cnt;
}
bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
memset(h, -1, sizeof h);
for(auto &entity:prerequisites) {
add(entity[1], entity[0]);
d[entity[0]] ++;
}
return topsort(numCourses);
}
};