Codeforces Round #513 - E. Sergey and Subway （树形DP）

http://codeforces.com/contest/1060/problem/E

题意：

问你一棵树上的所有点与点之间的距离。

不过可以2步当成1步走。（跳着走）（即如果两个点隔了一个点，就连一条边）

POINT：

首先，如果距离为偶数，直接除2，

如果距离为奇数，除2+1。

 

算出所有点与点之间的距离（不跳的真实距离）。树形DP解决。

可以算每条边被走了几次，边(u,v)被走了的次数=u的后面（前面）所有节点数*v的后面（前面）所有节点数。

实际上就是 一个是u的子树的大小(x)，和n-x。  一边和另一边嘛。很容易想通。对于所有边的和，就是距离之和ans了。

然后如果都是偶数距离，是不是直接除2就行了。但是我们有奇数。

奇数距离，是奇数层(深度deep) 和偶数层的点之间的。

所以再记录一下奇数层(y)有几个点就行了。剩下来的都是偶数点。

[ans-y*(n-y)]/2,这个为一直跳着走的距离。（减去y*(n-y) ，如果这些奇数多出来的1，加起来除2，答案就多了很多了）

然后加回来。

最终答案为[ans-y*(n-y)]/2+y*(n-y).

 

#include <vector>
#include <stdio.h>
#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;
#define LL long long
const int N = 2e5+5;
vector<int>G[N];

LL sz[N];
LL num=0;

void dfs(int u,int pre,int k)
{
	if(k) num++;
	sz[u]=1;
	for(int i=0;i<G[u].size();i++){
		int v=G[u][i];
		if(v==pre) continue;
		dfs(v,u,k^1);
		sz[u]+=sz[v];
	}
}

int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		G[u].push_back(v);G[v].push_back(u);
	}
	dfs(1,-1,1);
	LL ans=0;
	for(int i=2;i<=n;i++){
		ans+=sz[i]*(n-sz[i]);
	}
	ans=(ans-num*(n-num))/2+num*(n-num);
	printf("%lld\n",ans);



}