bzoj2152[国家集训队]聪聪可可 点分治

原题:https://www.lydsy.com/JudgeOnline/problem.php?id=2152

题解:本质上是求树上有多少3的倍数长度的路径。树上路径的个数考虑用点分治。对于每一次分治我们需要求出过一点3的倍数的长度的路径，不妨从这个点搜索，记上余数是0，1，2，的链数，为t[0],t[1],t[2]。两两组合并且可以颠倒，即t[0]*t[0]+2*t[1]*t[2]。转移时减掉重复的就行了。很套路的一道题。

#include<bits/stdc++.h>
#define inf (1<<31)-1
using namespace std;
const int N=20005;
struct E{int to,c,nxt;}data[N<<1]; 
int len=1,h[N],vis[N],n,f[N],sz[N],inv,fac,rt,d[N],t[10];

int gcd(int x,int y){return x==0?y:gcd(y%x,x);}

inline void ins(int x,int y,int c){
	data[++len].to=y;data[len].c=c;data[len].nxt=h[x];h[x]=len;
	data[++len].to=x;data[len].c=c;data[len].nxt=h[y];h[y]=len;
}

inline int rd(){
	int x=0;int f=1;char s=getchar();
	while(!isdigit(s))f=(s=='-'?-1:f),s=getchar();
	while(isdigit(s)) x=(x<<1)+(x<<3)+s-'0',s=getchar();
	return x*f;
}
void getroot(int x,int fa){
	sz[x]=1;f[x]=0;
	for(int i=h[x];i;i=data[i].nxt){
		int y=data[i].to;if(y==fa || vis[y])continue;
		getroot(y,x);
		sz[x]+=sz[y]; 
		f[x]=max(f[x],sz[y]);
	}
	f[x]=max(f[x],n-sz[x]);
	if(f[rt]>f[x]) rt=x;
}
void dfs(int x,int fa){
	t[d[x]%3]++;
	for(int i=h[x];i;i=data[i].nxt){
		int y=data[i].to;if(y==fa || vis[y]) continue;
		d[y]=(d[x]+data[i].c)%3;
		dfs(y,x);
	}	
}
int work(int x,int c){
	t[0]=t[1]=t[2]=0;
	d[x]=c;
	dfs(x,0);
	return t[0]*t[0]+t[1]*t[2]*2;
}
void sol(int x){
	
	fac+=work(x,0);
	vis[x]=1;
	for(int i=h[x];i;i=data[i].nxt){
		int y=data[i].to;if(vis[y]) continue;
		fac-=work(y,data[i].c);
		n=sz[y];rt=0;getroot(y,0);
		sol(rt);		
	}	
}

int main(){
//	freopen("bzoj2152.in","r",stdin);
	memset(vis,0,sizeof vis);
	memset(h,0,sizeof h);
	len=1;f[0]=inf;rt=0;fac=0;
	inv=n=rd();inv=inv*inv;
	for(int i=1,x,y,c;i<n;i++){
		x=rd();y=rd();c=rd();
		ins(x,y,c);
	} 
	getroot(1,0);
	sol(rt);
	int d=gcd(fac,inv);
	printf("%d/%d\n",fac/d,inv/d);
	
	return 0;
}