luogu P2634 聪聪可可(点分治)

题目链接

题目描述
聪聪和可可是兄弟俩，他们俩经常为了一些琐事打起来，例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑（可是他们家只有一台电脑）……遇到这种问题，一般情况下石头剪刀布就好了，可是他们已经玩儿腻了这种低智商的游戏。

他们的爸爸快被他们的争吵烦死了，所以他发明了一个新游戏：由爸爸在纸上画 $n$ 个“点”，并用 $n-1$ 条“边”把这 $n$ 个“点”恰好连通（其实这就是一棵树）。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点（当然他们选点时是看不到这棵树的），如果两个点之间所有边上数的和加起来恰好是 $3$ 的倍数，则判聪聪赢，否则可可赢。

聪聪非常爱思考问题，在每次游戏后都会仔细研究这棵树，希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。

输入格式
输入的第 $1$ 行包含 $1$ 个正整数 $n$。后面 $n-1$ 行，每行 $3$ 个整数 $x,y,w$，表示 $x$ 号点和 $y$ 号点之间有一条边，上面的数是 $w$。

输出格式
以即约分数形式输出这个概率（即 $a/b$ 的形式，其中 $a$ 和 $b$ 必须互质。如果概率为 $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);
}