子树的结点个数

题目描述

有一个棵树，树上有 n个结点。结点的编号分别为 1…n，其中 1 是树的根结点。
现在希望你帮忙计算每个结点作为根结点的子树分别有多少结点。

输入描述

第一行输入一个数字 n，代表树上结点的个数。（2≤n≤1000）接下来的 n−1 行，每行俩个数字 a，b，代表结点 a 到结点 b 有一条边。

输出描述

按编号顺序输出每个结点作为根结点的子树，分别有多少结点，中间用空格分开。

样例输入

5
1 4
1 3
3 2
3 5

样例输出

5 1 3 1 1

思路

见代码吧

AC代码

#include <iostream>
#include <string.h>
using namespace std;
const  int maxn = 20000;
struct edge{
    int v,next;
}e[maxn];
int p[maxn],eid;
void init(){
    memset(p,-1, sizeof(p));
    eid = 0;
}
void insert(int u,int v){
    e[eid].v = v;
    e[eid].next = p[u];
    p[u]  = eid++;
}
bool vst[maxn];
int cnt[maxn];//记录结点个数
void dfs(int u){
    vst[u] = true;
    cnt[u] = 1;//先把自身算进去
    for(int i = p[u];i!=-1;i= e[i].next){
        if(!vst[e[i].v]){
            dfs(e[i].v);**//这步加上cnt[u] = 1可以看成把全部结点的cnt都赋值为1；（按照样例）深搜1是可以深搜到所有的点的，所以1+加上cnt[2],cnt[3]...就是cnt[1]=5,4的邻接点都被深搜过了，所以cnt[4] = 1，3可以搜到2,5，所以cnt[3] = 3,2和5的邻接节点都被访问过了，所以cnt都是1；**
            cnt[u]+=cnt[e[i].v];
        }
    }
}
int main(){
    init();
    memset(vst,false,sizeof(vst));
    int n;
    cin>>n;
    int u,v;
    for(int i = 0;i<n-1;i++){
        cin>>u>>v;
        insert(u,v);
        insert(v,u);
    }
    dfs(1);
    for(int i = 1;i<=n;i++){
        cout<<cnt[i];
        if(i!=n) cout<<" ";
    }
    return 0;
}