bzoj 2152: 聪聪可可（树分治）

2152: 聪聪可可
Time Limit: 3 Sec Memory Limit: 259 MB
Submit: 5933 Solved: 3156
[Submit][Status][Discuss]
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

【样例说明】

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)。

【数据规模】

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

HINT

Source

[Submit][Status][Discuss]

#include<iostream>
#include<algorithm>
#include<string.h>
#include<stdio.h>
using namespace std;

#define N 20010
#define inf 1e9+10

struct node{
	int to,c,next;
}g[N*2]; 

int head[N],m;
int size[N],f[N];//f:最大子树节点数
bool vis[N];
int d[N],deep[N];
int t[4];
int n,sum,root,k,ans;

void add_edge(int from,int to,int cost){
	g[++m].next=head[from];
	head[from]=m;
	g[m].to=to;g[m].c=cost;
}

void getroot(int v,int fa){
	size[v]=1;f[v]=0;
	for(int i=head[v];i;i=g[i].next){
		if(g[i].to!=fa&&!vis[g[i].to]){
			getroot(g[i].to,v);
			size[v]+=size[g[i].to];
			f[v]=max(f[v],size[g[i].to]);
		}
	}
	f[v]=max(f[v],sum-size[v]);
	if(f[v]<f[root]) root=v;
}

void getdeep(int v,int fa){
	deep[++deep[0]]=d[v]%3;
	t[d[v]%3]++;	
	for(int i=head[v];i;i=g[i].next){
		if(g[i].to!=fa&&!vis[g[i].to]){
			d[g[i].to]=(d[v]+g[i].c)%3;
			getdeep(g[i].to,v);
		}
	}
}


int cal(int v,int cost){
	t[0]=t[1]=t[2]=0;
	d[v]=cost;
	deep[0]=0;
	getdeep(v,0);
	return t[1]*t[2]*2+t[0]*t[0];
}

void solve(int v){
	ans+=cal(v,0);
	vis[v]=1;
	for(int i=head[v];i;i=g[i].next){
		if(!vis[g[i].to]){
			ans-=cal(g[i].to,g[i].c);
			sum=size[g[i].to];
			root=0;
			getroot(g[i].to,0);
			solve(root);
		}
	}
}

int main(){
	int u,v,w;
	while(~scanf("%d",&n)&&n){
		ans=root=m=0;
		memset(vis,0,sizeof(vis));
		memset(head,0,sizeof(head));
		for(int i=1;i<n;i++){
			scanf("%d%d%d",&u,&v,&w);
			w=w%3;
			add_edge(u,v,w);
			add_edge(v,u,w);
		}
		f[0]=inf;
		sum=n;
		getroot(1,0);
		solve(root);
		int x=__gcd(ans,n*n);
		printf("%d/%d\n",ans/x,n*n/x);
	}
}