51nod 1405 樹的距離之和

給定一棵無根樹，假設它有n個節點，節點編號從1到n, 求任意兩點之間的距離（最短路徑）之和。

Input

第一行包含一個正整數n (n <= 100000)，表示節點個數。
後面(n - 1)行，每行兩個整數表示樹的邊。

Output

每行一個整數，第i(i = 1,2,...n)行表示所有節點到第i個點的距離之和。

Input示例

4
1 2
3 2
4 2

Output示例

5
3
5
5

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

樹形DP~

用siz[i]表示以i爲根的子樹的大小，f[i]表示所有點到i的距離之和，那麼求siz的時候順便求出f[1]，那麼如果k是u的子節點，那麼f[k]=f[u]-siz[k]+n-siz[k]，因爲子樹內的點距離k近1，其餘點遠1。

要開long long!

#include<cstdio>
#include<iostream>
using namespace std;
#define ll long long

int n,x,y,fi[100001],w[200001],ne[200001],cnt,siz[100001];
ll f[100001];

int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0' || ch>'9') {if(ch=='-') f=-1;ch=getchar();}
	while(ch>='0' && ch<='9') {x=x*10+ch-'0';ch=getchar();}
	return x*f;
}

void add(int u,int v)
{
	w[++cnt]=v;ne[cnt]=fi[u];fi[u]=cnt;
	w[++cnt]=u;ne[cnt]=fi[v];fi[v]=cnt;
}

ll dfs(int u,int fa)
{
	ll now=0;siz[u]=1;
	for(int i=fi[u];i;i=ne[i])
	  if(w[i]!=fa)
	  {
	  	now+=dfs(w[i],u);siz[u]+=siz[w[i]];
	  }
	return now+siz[u];
}

void dfs1(int u,int fa)
{
	for(int i=fi[u];i;i=ne[i])
	  if(w[i]!=fa)
	  {
	  	f[w[i]]=f[u]-siz[w[i]]+n-siz[w[i]];
	  	dfs1(w[i],u);
	  }
}

int main()
{
	n=read();
	for(int i=1;i<n;i++) x=read(),y=read(),add(x,y);
	f[1]=dfs(1,0)-siz[1];dfs1(1,0);
	for(int i=1;i<=n;i++) printf("%lld\n",f[i]);
	return 0;
}