51nod-1405 树的距离之和

思路：可以先随便选取节点1作为树的根节点，然后从根节点遍历树，每次保存k节点的子节点总数 d[k].s 以及所有子节点都k节点的距离和 d[k].sum，在遍历过程就可完成保存。

然后再从根节点遍历一遍树，求所有点到k节点的距离，就可以转换成求k节点的子节点到k节点的距离s1和k节点的头节点 id 其所在的另一树的节点到k节点的距离s2。

而s1=d[k].sum,s2=(头结点所在树的sum-k节点子树的sum-k节点的所有节点到从k节点到头结点的距离)+(头结点所在的另一颗树到k节点的距离)=(d[id].sum-d[k].sum-d[k].s)+(n-d[k].s); 所有点到k节点的距离d[k].sum=s1+s2=d[id].sum-d[c].s+n-d[c].s;

Code :

#include<iostream>
#include<vector>
using namespace std;
typedef long long LL;

struct node{
	int s;
	LL sum;
};
const int MAX_N=100005;
int n;
vector<int> e[MAX_N];
node d[MAX_N];

void Build(int id,int pre);
void DFS(int id,int pre);
int main()
{
	ios::sync_with_stdio(false);
	cin>>n;
	int u,v;
	for(int i=1;i<n;++i)
	{
		cin>>u>>v;
		e[u].push_back(v);
		e[v].push_back(u);
	}
	Build(1,0);
	DFS(1,0);
	for(int i=1;i<=n;++i)
		cout<<d[i].sum<<endl;
	
	return 0;
}

void Build(int id,int pre)
{
	LL s=1,sum=0;
	for(auto c:e[id])
		if(c!=pre){
			Build(c,id);
			s+=d[c].s;	sum+=d[c].sum+d[c].s;
		}
	d[id]=node{(int)s,sum};
}

void DFS(int id,int pre)
{
	for(auto c:e[id])
		if(c!=pre){
			d[c].sum=d[id].sum-d[c].s+n-d[c].s;
			DFS(c,id);
		}
}