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

2152: 聪聪可可

Time Limit: 3 Sec  Memory Limit: 259 MB
Submit: 3741  Solved: 1925
[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

和http://blog.csdn.net/jaihk662/article/details/78161182这题很像

不同的是这题权值和要对3取模，只要会树的点分治的话应该是没问题的

#include<stdio.h>#include<vector>using namespace std;typedef struct{int y;int len;}Point;Point now;vector<Point> G[20005];int ans, n, bet, heart, vis[20005], size[20005], dp[3], temp[3];int Gcd(int x, int y){if(y==0)return x;return Gcd(y, x%y);}void Sech1(int u, int p){int i, v;size[u] = 1;for(i=0;i<G[u].size();i++){v = G[u][i].y;if(v==p || vis[v])continue;Sech1(v, u);size[u] += size[v];}}void Sech2(int u, int p, int &psiz){int bsiz, v, i;bsiz = psiz-size[u];for(i=0;i<G[u].size();i++){v = G[u][i].y;if(v==p || vis[v])continue;bsiz = max(bsiz, size[v]);Sech2(v, u, psiz);}if(bsiz<bet){bet = bsiz;heart = u;}}void Sech(int u, int p, int len){int i, v;temp[len%3]++;for(i=0;i<G[u].size();i++){v = G[u][i].y;if(vis[v] || v==p)continue;Sech(v, u, len+G[u][i].len);}}void Count(int u){int i, j, v;dp[0] = 1;dp[1] = dp[2] = 0;for(i=0;i<G[u].size();i++){v = G[u][i].y;if(vis[v])continue;temp[0] = 0;temp[1] = temp[2] = 0;Sech(v, u, G[u][i].len);ans += (dp[0]*temp[0]+dp[1]*temp[2]+dp[2]*temp[1])*2;dp[0] = dp[0]+temp[0];dp[1] = dp[1]+temp[1];dp[2] = dp[2]+temp[2];}}void Divide(int u){int i, v;Sech1(u, 0);bet = 200005;Sech2(u, 0, size[u]);vis[heart] = 1;u = heart;Count(u);for(i=0;i<G[u].size();i++){v = G[u][i].y;if(vis[v])continue;Divide(v);}}int main(void){int x, y, len, i;//freopen("in.txt", "r", stdin);scanf("%d", &n);for(i=1;i<=n-1;i++){scanf("%d%d%d", &x, &y, &len);now.y = y, now.len = len;G[x].push_back(now);now.y = x;G[y].push_back(now);}ans = n;Divide(1);y = Gcd(ans, n*n);printf("%d/%d\n", ans/y, n*n/y);return 0;}