dfs 的两种写法
在看之前实现图的遍历 dfs 和拓扑排序 dfs 实现的代码的时候的感悟
图的遍历 dfs 和拓扑排序 dfs 的区别
0 → 1
↓ ↓
2 → 3
图的邻接表表示:
adjList[0] = {1, 2};
adjList[1] = {3};
adjList[2] = {3};
adjList[3] = {};
-
正常的 DFS 遍历:
从节点 0 开始执行 DFS,假设我们按照邻接表的顺序遍历:
从节点 0 开始,访问 0,然后访问 0 的邻接节点 1 和 2
从节点 1 开始,访问 1,然后访问 1 的邻接节点 3
从节点 3 开始,访问 3,发现没有邻接节点
回到节点 1,再回到节点 0,访问 2
从节点 2 开始,访问 2,然后访问 2 的邻接节点 3(但 3 已经访问过了)
DFS 遍历的顺序:0 -> 1 -> 3 -> 2
-
拓扑排序的 DFS 遍历:
在拓扑排序中,我们使用栈存储节点,并确保先访问所有邻接节点再加入栈
从节点 0 开始,访问 0,并标记为“正在访问”
访问 0 的邻接节点 1,并标记为“正在访问”
访问 1 的邻接节点 3,并标记为“正在访问”
节点 3 没有邻接节点,标记为“已访问”,并将 3 放入栈中
回到节点 1,标记为“已访问”,并将 1 放入栈中
回到节点 0,访问 2,并标记为“正在访问”
访问 2 的邻接节点 3(已访问过),因此直接标记 2 为“已访问”,并将 2 放入栈中
回到节点 0,标记为“已访问”,并将 0 放入栈中
拓扑排序的顺序:3 -> 1 -> 2 -> 0
总结两者的区别
DFS 遍历按照遇到的第一个未访问的邻接节点进行递归访问, 直接记录 节点的访问顺序
拓扑排序要求节点在加入结果时要等到其 所有邻接节点被完全访问 过,因此节点的加入顺序是倒序的(即后序遍历)
代码
原 dfs 遍历代码
void MyGraph::dfsVisit(int vertex, vector<bool>& visited, vector<int>& traversal)
{visited[vertex] = true;traversal.push_back(vertex);for (const auto& edge : adjList[vertex])if (!visited[edge.first])dfsVisit(edge.first, visited, traversal);
}void MyGraph::DFS(int startVertex)
{vector<bool> visited(n, false); vector<int> traversal; dfsVisit(startVertex, visited, traversal);for (int vertex : traversal) cout << vertex << " ";cout << endl;
}
拓扑排序的 dfs 实现
bool dfs(int node, vector<vector<int>> &adj, vector<int> &visited, stack<int> &result)
{// 当前节点正在访问,图中有环if (visited[node] == 1)return false; // 当前节点已经访问过if (visited[node] == 2)return true; // 1. 对每个未被访问的节点执行 DFS,标记该节点为访问中visited[node] = 1; for (int neighbor : adj[node])// 2. 如果访问到某个已经在 “ 访问中 ” 状态的节点,则说明图中存在环,无法进行拓扑排序if (!dfs(neighbor, adj, visited, result))return false;// 3. DFS 完成后,标记节点为“已访问”,并将节点加入到结果栈(或者队列)中visited[node] = 2; // 标记为已访问result.push(node); // 将节点加入栈return true;
}vector<int> topologicalSortDFS(int n, vector<vector<int>> &edges)
{vector<vector<int>> adj(n);for (const auto &edge : edges)adj[edge[0]].push_back(edge[1]);vector<int> visited(n, 0); // 0: 未访问, 1: 正在访问, 2: 已访问stack<int> result;// 1. 对每个未被访问的节点执行 DFS,标记该节点为访问中for (int i = 0; i < n; ++i)if (visited[i] == 0)if (!dfs(i, adj, visited, result))return {}; // 如果存在环,返回空vector<int> order;while (!result.empty()){order.push_back(result.top());result.pop();}return order;
}
想到如果同样对 dfs 遍历的返回值改为 bool 值也可以写成
bool MyGraph::dfsVisit(int vertex, vector<bool> &visited, vector<int> &traversal)
{if (visited[vertex])return true;visited[vertex] = true; traversal.push_back(vertex); for (const auto &edge : adjList[vertex])if (!dfsVisit(edge.first, visited, traversal))return false;return true;
}void MyGraph::DFS(int startVertex)
{vector<bool> visited(n, false); vector<int> traversal; if (dfsVisit(startVertex, visited, traversal)){cout << "DFS Traversal: ";for (int vertex : traversal){cout << vertex << " ";}cout << endl;}
}