bzoj2152 聪聪可可 (点分治)

问题描述

聪聪和可可是兄弟俩，他们俩经常为了一些琐事打起来，例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑（可是他们家只有一台电脑）……遇到这种问题，一般情况下石头剪刀布就好了，可是他们已经玩儿腻了这种低智商的游戏。他们的爸爸快被他们的争吵烦死了，所以他发明了一个新游戏：由爸爸在纸上画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。

思路：

点分治计算出所有点对距离模3分别等于0,1,2的数量
满足条件的答案就是sum[0]*sum[0]+sum[1]*sum[2]*2;//点对有序
！注意sum[0]不用*2

code：

#include<cstdio>
#include<cstring>
#include<vector>
#include<iostream>
#include<algorithm>
int read(){
    int x=0;
    char ch=getchar();
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return x;
}
using namespace std;
#define ll long long
const int maxm=2e4+5;
struct mp{
    int v,w;
    mp(int vv,int ww){
        v=vv,w=ww;
    }
};
vector<mp>g[maxm];
int sum[3];
int mark[maxm];
int sz[maxm],son[maxm];
int n,size,root;
ll ans;
void getroot(int x,int fa){
    sz[x]=1;
    son[x]=0;
    int len=g[x].size();
    for(int i=0;i<len;i++){
        int v=g[x][i].v;
        if(v==fa||mark[v])continue;
        getroot(v,x);
        sz[x]+=sz[v];
        son[x]=max(son[x],sz[v]);
    }
    son[x]=max(son[x],size-sz[x]);
    if(son[x]<son[root]){
        root=x;
    }
}
void getd(int x,int val,int fa){
    sum[val]++;
    int len=g[x].size();
    for(int i=0;i<len;i++){
        int v=g[x][i].v;
        int w=g[x][i].w;
        if(mark[v]||v==fa)continue;
        getd(v,(val+w)%3,x);
    }
}
ll solve(int x,int val){
    sum[0]=sum[1]=sum[2]=0;
    getd(x,val%3,0);
    int res=(ll)sum[0]*sum[0]+(ll)sum[1]*sum[2]*2;
    return res;
}
void divide(int x){
    ans+=solve(x,0);
    mark[x]=1;
    int len=g[x].size();
    for(int i=0;i<len;i++){
        int v=g[x][i].v;
        int w=g[x][i].w;
        if(mark[v])continue;
        ans-=solve(v,w);
        son[root=0]=size=sz[v];
        getroot(v,0);
        divide(root);
    }
}
ll gcd(ll a,ll b){
    return b==0?a:gcd(b,a%b);
}
signed main(){
    n=read();
    for(int i=1;i<n;i++){
        int a,b,c;
        a=read(),b=read(),c=read();
        g[a].push_back(mp(b,c));
        g[b].push_back(mp(a,c));
    }
    son[root=0]=size=n;
    getroot(1,0);
    divide(root);
    ll t=gcd(ans,n*n);
    printf("%lld/%lld\n",ans/t,n*n/t);
    return 0;
}