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
#include
using namespace std;
typedef long long LL;

struct node{
	int s;
	LL sum;
};
const int MAX_N=100005;
int n;
vector 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>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<