main
黄海 1 year ago
parent 6822f7dd8d
commit 3673870d36

@ -1,8 +1,8 @@
#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
vector<int> g[N], f[N]; // 邻接表,一个正图,一个反图
int c[N]; // 每个树洞中松鼠的数量
vector<int> g[N]; // 邻接表,一个正图,一个反图
int c[N]; // 每个树洞中松鼠的数量
/*
4
@ -15,19 +15,21 @@ int c[N]; // 每个树洞中松鼠的数量
2 4
2
*/
int dfs1(int u, int k) {
if (k == 0) return 0;
int dfs(int u, int fa, int k) {
if (k == 0) return c[u];
int sum = c[u];
for (int i = 0; i < g[u].size(); i++) sum += dfs1(g[u][i], k - 1);
return sum;
}
int dfs2(int u, int k) {
if (k == 0) return 0;
int sum = c[u];
for (int i = 0; i < f[u].size(); i++) sum += dfs2(f[u][i], k - 1);
for (int i = 0; i < g[u].size(); i++) {
if (g[u][i] == fa) continue;
sum += dfs(g[u][i], u, k - 1);
}
return sum;
}
int main() {
#ifndef ONLINE_JUDGE
freopen("11.in", "r", stdin);
#endif
int n; // 树洞的数量
cin >> n;
@ -37,21 +39,11 @@ int main() {
int a, b; // 表示两个树洞相连接
cin >> a >> b;
g[a].push_back(b); // 用邻接表保存一下某个树洞与哪两个其它树洞相连接
f[b].push_back(a); // 建反图
g[b].push_back(a);
}
int k;
cin >> k;
for (int i = 1; i <= n; i++) {
// 计算i号树洞与它在k个距离内的所有树洞的松鼠个数和
// 1、向下k个距离 k<=20
int sum = dfs1(i, k);
// 2、向上k个距离 k<=20
sum += dfs2(i, k);
cout << sum - c[i] << endl;
}
for (int i = 1; i <= n; i++) cout << dfs(i, -1, k) << endl;
return 0;
}
Loading…
Cancel
Save