BSOJ3991：聪聪可可 点分治

Description
　　聪聪和可可是兄弟俩，他们俩经常为了一些琐事打起来，例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑（可是他们家只有一台电脑）……遇到这种问题，一般情况下石头剪刀布就好了，可是他们已经玩儿腻了这种低智商的游戏。他们的爸爸快被他们的争吵烦死了，所以他发明了一个新游戏：
　　由爸爸在纸上画n个“点”，并用n-1条“边”把这n个“点”恰好连通（其实这就是一棵树）。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点（当然他们选点时是看不到这棵树的），如果两个点之间所有边上数的和加起来恰好是3的倍数，则判聪聪赢，否则可可赢。
　　聪聪非常爱思考问题，在每次游戏后都会仔细研究这棵树，希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。
Input
输入的第1行包含1个正整数n。
后面n-1行，每行3个整数x、y、w，表示x号点和y号点之间有一条边，上面的数是w。
Output
以即约分数形式输出这个概率（即“a/b”的形式，其中a和b必须互质。如果概率为1，输出“1/1”）。
Sample Input
5
1 2 1
1 3 2
1 4 1
2 5 3
Sample Output
13/25
Hint
【样例说明】
　　13组点对分别是(1,1) (2,2) (2,3) (2,5) (3,2) (3,3) (3,4) (3,5) (4,3) (4,4) (5,2) (5,3) (5,5)。


【数据规模】
　　对于30%的数据，n<=1000；
　　另有20%的数据，给出的树中每个节点的度不超过2；

　　对于100%的数据，n<=20000。

最简单的点分治，建边时把边权mod 3，加值的时候也mod 3，就可以解决了

下面是注释和代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define size 400005
using namespace std;
struct Edge
{
	int to,next,v;
}w[size*4];
int h[size]={0},cnt=0;
int n,d[size]={0};
bool vis[size]={0};
int hash[5]={0};
int Size[size]={0};
int gcd(int a,int b)
{  
   if(b==0)return a;
   else return gcd(b,a%b);
}

void add(int x,int y,int v)
{
	cnt++;w[cnt].to=y;w[cnt].next=h[x];h[x]=cnt;w[cnt].v=v;
}
void init()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		add(a,b,c%3);
		add(b,a,c%3);
	}
}
void DFS2(int x,int fa)
{
	Size[x]=1;
	for(int i=h[x];i;i=w[i].next)
	{
		int to=w[i].to;
		if(!vis[to]&&to!=fa)
		{
			DFS2(to,x);
			Size[x]+=Size[to];
		}
	}
}
int Findcenter(int x)
{
	DFS2(x,0);
	int c=x,tot=Size[x];
	while(1)
	{
		int cc=0;
		for(int i=h[c];i;i=w[i].next)
		{
			int to=w[i].to;
			if(!vis[to]&&Size[to]>Size[cc])cc=to;
		}
		if(Size[cc]<=tot/2)break;
		else
		{
			Size[c]-=Size[cc];
			Size[cc]=tot;
			c=cc;
		}
	}
	return c;
}
void Getdep(int x,int fa)
{
    hash[d[x]]++;
	for(int i=h[x];i;i=w[i].next)
	{
		int to=w[i].to;
		if(!vis[to]&&to!=fa)
		{
			d[to]=(d[x]+w[i].v)%3;
			Getdep(to,x);
		}	
	}
}
int Calc(int x,int len)
{
	hash[0]=hash[1]=hash[2]=0;
	d[x]=len%3;
	Getdep(x,0);
    return hash[0]*hash[0]+hash[1]*hash[2]*2;	

/*

一共选两边，<span style="font-family: Arial, Helvetica, sans-serif;">边权已经每步mod 3，现在要得到3(取余为0)，也就是被3整除，只能为0+0或者1+2或2+1，乘法原理+加法原理</span>

<span style="font-family: Arial, Helvetica, sans-serif;">*/</span>

}
int ans=0;
void DFS(int x)
{
	int G=Findcenter(x);
	vis[G]=1;
	ans+=Calc(G,0);
	for(int i=h[G];i;i=w[i].next)
	{
		int to=w[i].to;
		if(!vis[to])ans-=Calc(to,w[i].v),DFS(to);
	}
}
int main(){
    init();
    DFS(1);
    int GCD=gcd(ans,n*n);
    printf("%d/%d\n",ans/GCD,n*n/GCD);
	return 0;
}

代码来自hzwer