C++语言实现拓扑排序(Topological Sorting)算法
拓扑排序是一种用于有向无环图(DAG, Directed Acyclic Graph)的线性排序。它保证对于图中的每一条有向边 (u, v),顶点 u 在排序中出现在顶点 v 之前。下面是使用C++实现拓扑排序的两种常见方法:Kahn算法和深度优先搜索(DFS)算法。
使用Kahn算法实现拓扑排序
Kahn算法通过不断移除入度为0的节点来实现拓扑排序。
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
// 拓扑排序函数,返回拓扑排序的结果
vector<int> topologicalSortKahn(int n, vector<vector<int>>& adj) {
vector<int> in_degree(n, 0); // 记录每个节点的入度
vector<int> topo_order; // 存储拓扑排序结果
// 计算每个节点的入度
for (int i = 0; i < n; ++i) {
for (int j : adj[i]) {
++in_degree[j];
}
}
// 将入度为0的节点入队
queue<int> q;
for (int i = 0; i < n; ++i) {
if (in_degree[i] == 0) {
q.push(i);
}
}
// Kahn算法
while (!q.empty()) {
int u = q.front();
q.pop();
topo_order.push_back(u);
for (int v : adj[u]) {
--in_degree[v];
if (in_degree[v] == 0) {
q.push(v);
}
}
}
// 如果排序结果包含的节点数不等于图中的节点数,说明图中存在环
if (topo_order.size() != n) {
return {}; // 返回空数组表示失败
}
return topo_order;
}
int main() {
int n = 6; // 节点数
vector<vector<int>> adj = {
{2, 3}, // 节点0的邻接表
{3, 4}, // 节点1的邻接表
{4}, // 节点2的邻接表
{5}, // 节点3的邻接表
{5}, // 节点4的邻接表
{} // 节点5的邻接表
};
vector<int> result = topologicalSortKahn(n, adj);
if (result.empty()) {
cout << "图中存在环,无法进行拓扑排序。" << endl;
} else {
cout << "拓扑排序结果:" << endl;
for (int i : result) {
cout << i << " ";
}
cout << endl;
}
return 0;
}