题目描述
聪聪和可可是兄弟俩,他们俩经常为了一些琐事打起来,例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑(可是他们家只有一台电脑)……遇到这种问题,一般情况下石头剪刀布就好了,可是他们已经玩儿腻了这种低智商的游戏。
他们的爸爸快被他们的争吵烦死了,所以他发明了一个新游戏:由爸爸在纸上画 n n n 个“点”,并用 n − 1 n-1 n−1 条“边”把这 n n n 个“点”恰好连通(其实这就是一棵树)。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点(当然他们选点时是看不到这棵树的),如果两个点之间所有边上数的和加起来恰好是 3 3 3 的倍数,则判聪聪赢,否则可可赢。
聪聪非常爱思考问题,在每次游戏后都会仔细研究这棵树,希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。
输入格式
输入的第
1
1
1 行包含
1
1
1 个正整数
n
n
n。后面
n
−
1
n−1
n−1 行,每行
3
3
3 个整数
x
,
y
,
w
x,y,w
x,y,w,表示
x
x
x 号点和
y
y
y 号点之间有一条边,上面的数是
w
w
w。
输出格式
以即约分数形式输出这个概率(即
a
/
b
a/b
a/b 的形式,其中
a
a
a 和
b
b
b 必须互质。如果概率为
1
1
1,输出
1
/
1
1/1
1/1)。
输入 #1
5
1 2 1
1 3 2
1 4 1
2 5 3
输出 #1
13/25
解法: 点分治模板题
Code:
#include <bits/stdc++.h>
using namespace std;
const int MX = 2e4 + 7;
ll gcd(ll a,ll b){return b ? gcd(b,a%b):a;}
struct Edge{
int v,w,next;
}e[MX << 1];
int ecnt,head[MX];
void add(int u,int v,int w){
e[++ecnt].v = v;
e[ecnt].w = w;
e[ecnt].next = head[u];
head[u] = ecnt;
}
int n,sum,cnt,res,rt;
int siz[MX],maxp[MX],tmp[MX],dis[MX],judge[3];
bool vis[MX];
void getrt(int u,int fa){
siz[u] = 1, maxp[u] = 0;
for(int i = head[u];i;i = e[i].next){
int v = e[i].v;
if(vis[v] || v == fa) continue;
getrt(v,u);
siz[u] += siz[v];
maxp[u] = max(maxp[u],siz[v]);
}
maxp[u] = max(maxp[u], sum - siz[u]);
if(maxp[u] < maxp[rt]) rt = u;
}
void getdis(int u,int fa){
tmp[cnt++] = dis[u] % 3;
for(int i = head[u];i;i = e[i].next){
int v = e[i].v;
if(v == fa || vis[v]) continue;
dis[v] = dis[u] + e[i].w;
getdis(v,u);
}
}
void solve(int u){
queue<int>q;
for(int i = head[u];i;i = e[i].next){
int v = e[i].v;
if(vis[v]) continue;
cnt = 0;
dis[v] = e[i].w;
getdis(v,u);
for(int j = 0;j < cnt;++j){
if(tmp[j] == 0){
res+=2;
if(judge[0]) res += judge[0] * 2;
} else if(tmp[j] == 1){
if(judge[2]) res += judge[2] * 2;
} else if(tmp[j] == 2){
if(judge[1]) res += judge[1] * 2;
}
}
for(int j = 0;j < cnt;++j)
q.push(tmp[j]), judge[tmp[j]]++;
}
while(!q.empty()){
judge[q.front()] = 0; q.pop();
}
}
void divide(int u){
vis[u] = true;
solve(u);
for(int i = head[u];i;i = e[i].next){
int v = e[i].v;
if(vis[v]) continue;
maxp[rt=0] = sum = siz[v];
getrt(v,0);getrt(rt,0);
divide(rt);
}
}
void init(){
res = ecnt = rt = sum = cnt = 0;
clr(judge,0);
for(int i = 0;i <= n;++i){
head[i] = maxp[i] = dis[i] = siz[i] = 0;
vis[i] = false;
}
}
int main(){
scanf("%d",&n);
for(int i = 1;i < n;++i){
int u,v,w;scanf("%d %d %d",&u,&v,&w);
add(u,v,w);add(v,u,w);
}
maxp[0] = sum = n;
getrt(1,0);
getrt(rt,0);
divide(rt);
res += n;
int a = res, b = n * n;
int g = gcd(a,b);
printf("%d/%d\n", a/g,b/g);
}