51nod 1405 树的距离之和

#include <bits/stdc++.h>
using namespace std;

const int MAXN=100100;
vector<int> eg[MAXN];
long long sonsum[MAXN],dp[MAXN];
long long n;

long long dfs1(int x,int fa)
{
	long long ret=0;
	sonsum[x]=1;
	for(int i=0;i<eg[x].size();i++)
	{
		if(eg[x][i]==fa)
			continue;
		ret+=dfs1(eg[x][i],x);
		sonsum[x]+=sonsum[eg[x][i]];
	}
	ret+=sonsum[x]-1;
	return ret;
}

void dfs2(int x,int fa)
{
	for(int i=0;i<eg[x].size();i++)
	{
		if(eg[x][i]==fa)
			continue;
		dp[eg[x][i]]=dp[x]+n-2*sonsum[eg[x][i]];
		dfs2(eg[x][i],x);
	}
}

int main()
{
	int i,u,v;
	while(cin>>n)
	{
		for(i=1;i<=n;i++)
			eg[i].clear();
		for(i=1;i<n;i++)
		{
			scanf("%d%d",&u,&v);
			eg[u].push_back(v);
			eg[v].push_back(u);
		} 
		dp[1]=dfs1(1,0);
		dfs2(1,0);
		for(i=1;i<=n;i++)
			printf("%lld\n",dp[i]);
	}
}