LeetCode第 207 题:课程表(C++)_zj-CSDN博客的进阶
输出一种顺序即可
「拓扑排序」的一个附加效果是:能够顺带检测有向图中是否存在环
bfs
class Solution {
public:
vector<int> res;
vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
//依赖关系抽象成有向图
vector<vector<int>> adj(numCourses);//邻接表
vector<int> indegree(numCourses);//入度
for(const auto &v : prerequisites){
adj[v[1]].push_back(v[0]);//v[1]指向v[0]
++indegree[v[0]];
}
queue<int> q;
for(int i = 0; i < numCourses; ++i){//起点的入度为0,任意一个入度为0的都能作为起点
if(indegree[i] == 0) q.push(i);
}
while(!q.empty()){
int v = q.front();
res.push_back(v);
q.pop();
for(const auto &w : adj[v]){
--indegree[w];
if(indegree[w] == 0) q.push(w);//当该门课的先决条件都已达成
}
}
return res.size() == numCourses ? res : vector<int>();
}
};
dfs
参考课程表 II - 课程表 II - 力扣(LeetCode),可以看官方题解的图。
其实就是检测是否有环的存在
class Solution {
public:
vector<int> res;
vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
vector<vector<int>> adj(numCourses);
vector<int> flags(numCourses, 0);
for(const auto &v : prerequisites) adj[v[1]].push_back(v[0]);
for(int i = 0; i < numCourses; ++i){//检查顶点i出发是否有环
if(flags[i] == 0 && !dfs(i, adj, flags)) return vector<int>();
}
reverse(res.begin(), res.end());//需要逆序
return res;
}
bool dfs(int i, vector<vector<int>> &adj, vector<int> &flags){
if(flags[i] == -1) return true; // 说明当前节点已被其他节点启动的 dfs 访问,无需重复搜索,返回 true
if(flags[i] == 1) return false;// 说明在本轮 dfs 中节点 i 被第 2 次访问,即有环,直回 false
flags[i] = 1;
for(const auto &v : adj[i]){
if(!dfs(v, adj, flags)) return false;
}
flags[i] = -1;//走到这一步说明顶点i的邻接点寻找下去都不存在环
res.push_back(i);
return true;
}
};
后者也可以定义一个全局变量来记录是否存在环。
或者dfs还可以通过逆邻接表进行遍历,不过原理是一样的:对于一个结点来说,先输出指向它的所有顶点,再输出自己。