输入格式
第一行包含整数 n,表示树的结点数。
接下来n-1行,每行包含两个整数α和b,表示点a和点b之间存在一条边。
输出格式
输出一个长度为n的sz序列,表示树上节点1~n所代表子树的大小。
数据范围
1≤n≤10^5
输入样例
9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6
输出样例:
9 3 2 4 1 1 1 1 1
运用AcWing 846. 树的重心的思想,并稍作修改即可。
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
int n;
int h[N], ne[N<<1], e[N<<1], idx, sz[N];
bool st[N];
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
int dfs(int u)
{
sz[u] = 1, st[u] = true;
for(int i=h[u]; ~i; i=ne[i])
{
int j = e[i];
if(!st[j])
{
int s = dfs(j);
sz[u] += s;
}
}
return sz[u];
}
int main()
{
cin>>n;
memset(h, -1, sizeof h);
for(int i=0; i<n-1; ++i)
{
int a, b;
scanf("%d%d", &a, &b);
add(a, b), add(b, a);
}
dfs(1);
for(int i=1; i<=n; ++i) printf("%d ", sz[i]);
puts("");
return 0;
}