【BZOJ 2152】聪聪可可 点分治

直接分治，然后dis对3取模，有且只有取模后为（1,2）或者（0，0）对答案有贡献加起来就好了

#include<cstdio>
#include<cstring>
#include<iostream>
#define maxn 20020
using namespace std;
int s[maxn],rt,size,f[maxn],n,m,dis[maxn],head[maxn],tot=1,vis[maxn],ans;
int t[3];
struct edge{int v,next,w;}e[maxn*2];
void adde(int a,int b,int c){e[tot].v=b,e[tot].w=c,e[tot].next=head[a];head[a]=tot++;}
int gcd(int a,int b){return b==0?a:gcd(b,a%b);}

void getrt(int u,int fa){
	f[u]=0;s[u]=1;
	for(int v,i=head[u];i;i=e[i].next){
		if(vis[v=e[i].v]||v==fa)continue;
		getrt(v,u);
		s[u]+=s[v];
		f[u]=max(f[u],s[v]);
	}
	f[u]=max(f[u],size-f[u]);
	if(f[u]<f[rt])rt=u;
}

void dfs(int u,int fa){
	t[dis[u]]++;
	for(int i=head[u],v;i;i=e[i].next){
		if(vis[v=e[i].v]||v==fa)continue;
		dis[v]=(dis[u]+e[i].w)%3;
		dfs(v,u);
	}
}

int calc(int u,int h){
	dis[u]=h;
	t[1]=t[2]=t[0]=0;
	dfs(u,u);
	return t[1]*t[2]*2+t[0]*t[0];
}

void solve(int u){
	vis[u]=1;
	ans+=calc(u,0);
	for(int v,i=head[u];i;i=e[i].next){
		if(vis[v=e[i].v])continue;
		ans-=calc(v,e[i].w);
		f[rt=0]=size=s[v];
		getrt(v,u);
		solve(rt);
	}
}

int main(){
	scanf("%d",&n);
	for(int a,b,c,i=1;i<n;i++){
		scanf("%d%d%d",&a,&b,&c);c%=3;
		adde(a,b,c),adde(b,a,c);
	}
	f[rt=0]=size=n;
	getrt(1,1);
	solve(rt);
	int g=gcd(ans,n*n);
	printf("%d/%d",ans/g,n*n/g);
	return 0;
}