题目描述
有一个棵树,树上有 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;
}