题目大意
给出一个
n
n
n个顶点
m
m
m条边的有向图,还有起点和终点,求出起点和终点之间的路径条数,并判断从起点出发的所有路径是都能到达终点。
解题思路
首先,如果求出两点间的路径数目完全可以如下一个dfs搞定:
void dfs(int u) {
if (u == b) {
ans++;
return;
}
for (auto v : G[u]) {
dfs(v);
}
}
但是无法判断是否从 a a a出发的所有路径都只到 b b b,这时利用dfs回溯的思想,初始值设置 c n t [ b ] = 1 cnt[b]=1 cnt[b]=1代表从 b b b到 b b b的路径只有一条,然后在其他点达到 b b b之后不断回溯,最后 c n t [ a ] cnt[a] cnt[a]的值就是两点间的路径种类数。但是这样也就是解决了路径总数的问题。实际上,如果一个点的 c n t cnt cnt值为 0 0 0而且在dfs的时候访问过,显然就不是只到 b b b,那么只需要枚举所有的点然后判断一下。
#include <bits/stdc++.h>
#include <unordered_map>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
const int maxn = 1e6 + 10;
vector<int> G[maxn];
bitset<maxn> vis;
int cnt[maxn];
int ans = 0;
int n, m, a, b;
// void dfs(int u) {
// if (u == b) {
// ans++;
// return;
// }
// for (auto v : G[u]) {
// dfs(v);
// }
// }
int dfs(int u) {
vis[u] = 1;
if (cnt[u]) return cnt[u];
for (auto v : G[u]) cnt[u] += dfs(v);
return cnt[u];
}
int main() {
cin >> n >> m;
for (int i = 1, u, v; i <= m; i++) {
cin >> u >> v;
G[u].push_back(v);
}
cin >> a >> b;
cnt[b] = 1;
dfs(a);
bool ok = 1;
for (int i = 1; i <= n; i++) {
if (vis[i] && !cnt[i]) {
ok = 0;
break;
}
}
cout << cnt[a] << " " << (ok ? "Yes" : "No") << endl;
return 0;
}