BZOJ2152 聪聪可可（点分治）

【题解】

基础的点分治题

还是分为路径是否经过root来讨论，计算点到根的距离时，一边计算，一边统计距离对3取模等于0,1,2的点的个数 

注意：统计答案时(a,a)算1次，(a,b)算2次 

【代码】

#include<stdio.h>
#include<stdlib.h>
int v[40005]={0},w[40005]={0},first[20005]={0},next[40005]={0};
int size[20005]={0},maxsize[20005]={0},vis[20005]={0},d[20005]={0},cnt[5];
int e=0,root=0,sum,ans=0;
int max(int a,int b)
{
	if(a>b) return a;
	return b;
}
void tj(int x,int y,int z)
{
	v[++e]=y;
	w[e]=z;
	next[e]=first[x];
	first[x]=e;
}
void getroot(int x,int fa)
{
	int i;
	size[x]=1;
	maxsize[x]=0;
	for(i=first[x];i!=0;i=next[i])
		if(v[i]!=fa&&vis[v[i]]==0)
		{
			getroot(v[i],x);
			size[x]+=size[v[i]];
			maxsize[x]=max(maxsize[x],size[v[i]]);
		}
	maxsize[x]=max(maxsize[x],sum-size[x]);
	if(root==0||maxsize[root]>maxsize[x]) root=x;
}
void getd(int x,int fa)
{
	int i;
	cnt[d[x]%3]++;
	for(i=first[x];i!=0;i=next[i])
		if(v[i]!=fa&&vis[v[i]]==0)
		{
			d[v[i]]=d[x]+w[i];
			getd(v[i],x);
		}
}
int getcnt(int x,int now)
{
	int i;
	d[x]=now;
	cnt[0]=cnt[1]=cnt[2]=0;
	getd(x,0);
	return cnt[0]*(cnt[0]-1)+cnt[1]*cnt[2]*2;
}
void work(int x)
{
	int i;
	vis[x]=1;
	ans+=getcnt(x,0);
	for(i=first[x];i!=0;i=next[i])
		if(vis[v[i]]==0)
		{
			ans-=getcnt(v[i],w[i]);
			root=0;
			sum=size[v[i]];
			getroot(v[i],x);
			work(root);
		}
}
int gcd(int a,int b)
{
	if(b==0) return a;
	return gcd(b,a%b);
}
int main()
{
	int n,i,x,y,z;
	scanf("%d",&n);
	for(i=1;i<n;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		tj(x,y,z);
		tj(y,x,z);
	}
	sum=n;
	getroot(1,0);
	work(root);
	ans+=n;
	printf("%d/%d",ans/gcd(ans,n*n),n*n/gcd(ans,n*n));
	return 0;
}