dfs深度优先遍历 求树上各节点代表子树大小

输入格式

第一行包含整数 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;
}