【bzoj2152】聪聪可可 点分治

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=2152

【题解】

还是点分治的板子。

统计答案的方法值得一学：用t数组记录连通块中所有点的距离对3取模后的结果，然后当前的答案就是t[0]*t[0]+t[1]*t[2]*2，为什么要乘2呢？——因为同一个点对（u，v）聪聪选u可可选v与聪聪选v可可选u是不一样的。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
#define FILE "read"
#define MAXN 20010
#define INF 1000000000
#define up(i,j,n) for(int i=j;i<=n;++i)
#define dn(i,j,n) for(int i=j;i>=n;--i)
#define cmax(a,b) a=max(a,b)
#define cmin(a,b) a=min(a,b)
namespace INIT{
	char buf[1<<15],*fs,*ft;
	inline char getc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline int read(){
		int x=0,f=1;  char ch=getc();
		while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getc();}
		while(isdigit(ch))  {x=x*10+ch-'0';  ch=getc();}
		return x*f;
	}
}using namespace INIT;
struct node{int y,next,v;}e[MAXN*2];
int n,len,ans1,ans2,root,sum,Link[MAXN],t[MAXN],dis[MAXN],size[MAXN],vis[MAXN],f[MAXN];
void insert(int x,int y,int v){e[++len].next=Link[x];Link[x]=len;e[len].y=y;e[len].v=v;}
int gcd(int a,int b){return b==0?a:gcd(b,a%b);}
void getroot(int x,int fa){
	f[x]=0;  size[x]=1;
	for(int i=Link[x];i;i=e[i].next){
		if(vis[e[i].y]||e[i].y==fa)  continue;
		getroot(e[i].y,x);  size[x]+=size[e[i].y];
		cmax(f[x],size[e[i].y]);
	}
	cmax(f[x],sum-size[x]);
	if(f[x]<f[root])  root=x;
}
void getdeep(int x,int fa){
	t[dis[x]]++;
	for(int i=Link[x];i;i=e[i].next){
		if(e[i].y==fa||vis[e[i].y])  continue;
		dis[e[i].y]=(dis[x]+e[i].v)%3;
		getdeep(e[i].y,x);
	}
}
int cal(int x,int v){
	dis[x]=v;  t[0]=t[1]=t[2]=0;  getdeep(x,0);
	return t[1]*t[2]*2+t[0]*t[0];
}
void solve(int x){
	ans1+=cal(x,0);  vis[x]=1;
	for(int i=Link[x];i;i=e[i].next){
		if(vis[e[i].y])  continue;
		ans1-=cal(e[i].y,e[i].v);
		sum=size[e[i].y]; root=0;
		getroot(e[i].y,0);  solve(root);
	}
}
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	n=sum=read();  ans2=n*n;  f[0]=INF;
	up(i,1,n-1) {int x=read(),y=read(),v=read()%3;insert(x,y,v);insert(y,x,v);}
	getroot(1,0); solve(root);
	int d=gcd(ans1,ans2);
	printf("%d/%d\n",ans1/d,ans2/d);
	return 0;
}