2018.10.04【CodeForces1060】E. Sergey and Subway（树形DP）

版权声明：转载请声明出处，谢谢配合。 https://blog.csdn.net/zxyoi_dreamer/article/details/82941253

传送门

---

解析:

首先我们把树按照黑白交错染色，显然黑点的新边只会连在黑点上，白点同理。

然后我们考虑统计每条边的贡献。

是的，我们不分点统计贡献，而是分边统计贡献。

考虑有几个点对会经过这条边就行了，反正每条边的边权是1。

最后还要统计一下黑白点的贡献。

显然贡献的统计被重复了两次，最后输出答案的时候要除2。

---

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc getchar
#define pc putchar
#define cs const

inline
ll getint(){
	re ll num;
	re char c;
	while(!isdigit(c=gc()));num=c^48;
	while(isdigit(c=gc()))num=(num<<1)+(num<<3)+(c^48);
	return num;
}

cs int N=200005;
int n;
int last[N],nxt[N<<1],to[N<<1],ecnt;
inline void addedge(int u,int v){
	nxt[++ecnt]=last[u],last[u]=ecnt,to[ecnt]=v;
	nxt[++ecnt]=last[v],last[v]=ecnt,to[ecnt]=u;
}

int siz[N];
int cnt[2];
ll ans=0;
inline
void dfs(int u,int fa,bool f){
	siz[u]=1;
	++cnt[f];
	for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]]){
		if(v==fa)continue;
		dfs(v,u,!f);
		siz[u]+=siz[v];
	}
	ans+=1ll*siz[u]*(n-siz[u]);
}

signed main(){
	n=getint();
	for(int re i=1;i<n;++i){
		int u=getint(),v=getint();
		addedge(u,v);
	}
	dfs(1,0,0);
	cout<<((ans+1ll*cnt[0]*cnt[1])>>1);
	return 0;
}