计蒜客 子树的结点个数

问题描述
有一个棵树，树上有 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 MAX_N=1010;
//邻接表的存储结构
struct edge{
	int v,next;
}e[MAX_N];
int p[MAX_N],eid;
int cnt[MAX_N];
int vst[MAX_N]={0};
//初始化 
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++;
}
int n;
void dfs(int u)
{
	vst[u]=1;
	for(int i=p[u];i+1;i=e[i].next)
	{
		if(!vst[e[i].v])
		{
			dfs(e[i].v); 
			cnt[u]+=cnt[e[i].v];
			//cout<<"e[i].v="<<e[i].v<<endl;
		    //cout<<cnt[e[i].v]<<endl;
		}
		
	}
	
}
int main()
{
	init();
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cnt[i]=1;
	}
	int u,v;
	for(int i=1;i<n;i++)
	{
		cin>>u>>v;
		insert(u,v);
	}
	dfs(1);
	for(int i=1;i<=n;i++)
	{
		if(i!=1)
		  cout<<" ";
		cout<<cnt[i];
	}
	cout<<endl;
	return 0;
}