首页 > 其他 > 详细

子树的结点个数

时间:2018-06-09 00:25:49      阅读:289      评论:0      收藏:0      [点我收藏+]

有一个棵树,树上有 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


#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5+10;
const int INF = 0x3f3f3f3f;
vector<int> G[maxn];
int dfs(int cur) //dfs(cur)——>搜索cur为根的子树的结点
{
    int cnt=1;
    for(int i=0;i<G[cur].size();i++)
    {
        cnt += dfs(G[cur][i]); //计算子数结点,cnt累加子孙点
    }
    return cnt;
}
int main()
{
    int u,v;
    int n;
    cin>>n;
    for(int i=1;i<n;i++)
    {
        cin>>u>>v;
        G[u].push_back(v); //有向边
    }
    for(int i=1;i<=n;i++)
    {
        cout<<dfs(i)<<' '; //1~n每个结点作为根结点的子树,分别有多少结点
    }
}

子树的结点个数

原文:https://www.cnblogs.com/Roni-i/p/9158088.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!