bzoj2152 聪聪可可 点分治

Description

---

聪聪和可可是兄弟俩，他们俩经常为了一些琐事打起来，例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑（可是他们家只有一台电脑）……遇到这种问题，一般情况下石头剪刀布就好了，可是他们已经玩儿腻了这种低智商的游戏。他们的爸爸快被他们的争吵烦死了，所以他发明了一个新游戏：由爸爸在纸上画n个“点”，并用n-1条“边”把这n个“点”恰好连通（其实这就是一棵树）。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点（当然他们选点时是看不到这棵树的），如果两个点之间所有边上数的和加起来恰好是3的倍数，则判聪聪赢，否则可可赢。聪聪非常爱思考问题，在每次游戏后都会仔细研究这棵树，希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。

数据规模

---

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

Solution

---

本蒟蒻第一题点分治
本来想着用树形dp做的，结果不太会就看了一波点分治的做法

所谓点分治就是求一类树上的路径问题，把路径分为经过根与不经过根的进行分治。为了保证复杂度需要每次找到子树的重心作为子树的一个根来处理更小规模的问题

需要注意的是形如(x,x)这样的点对只能算一次

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
#define max(x,y) ((x)>(y)?(x):(y))
#define fill(x,t) memset(x,t,sizeof(x))
const int INF=0x3f3f3f3f;
const int N=20005;
const int E=50005;
struct edge{int x,y,w,next;}e[E];
int ls[N],edCnt=0;
int mx_son[N],size[N],dep[N],vis[N],cnt[3];
int root,n,sum=n,ans=0;
int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x*v;
}
void addEdge(int x,int y,int w) {
    e[++edCnt]=(edge){x,y,w%3,ls[x]}; ls[x]=edCnt;
    e[++edCnt]=(edge){y,x,w%3,ls[y]}; ls[y]=edCnt;
}
int gcd(int x,int y) {return !y?x:gcd(y,x%y);}
void get_root(int now,int fa) {
    size[now]=1;
    mx_son[now]=0;
    for (int i=ls[now];i;i=e[i].next) {
        if (e[i].y==fa||vis[e[i].y]) continue;
        get_root(e[i].y,now);
        size[now]+=size[e[i].y];
        mx_son[now]=max(mx_son[now],size[e[i].y]);
    }
    mx_son[now]=max(mx_son[now],sum-size[now]);
    if (mx_son[now]<mx_son[root]) root=now;
}
void get_size(int now,int fa) {
    size[now]=1;
    for (int i=ls[now];i;i=e[i].next) {
        if (e[i].y==fa||vis[e[i].y]) continue;
        get_size(e[i].y,now);
        size[now]+=size[e[i].y];
    }
}
void get_dep(int now,int fa) {
    cnt[dep[now]]++;
    for (int i=ls[now];i;i=e[i].next) {
        if (e[i].y==fa||vis[e[i].y]) continue;
        dep[e[i].y]=(dep[now]+e[i].w)%3;
        get_dep(e[i].y,now);
    }
}
int cal(int x,int y) {
    dep[x]=y;
    cnt[0]=cnt[1]=cnt[2]=0;
    get_dep(x,0);
    return cnt[0]*cnt[0]+cnt[1]*cnt[2]*2;
}
void solve(int now) {
    ans+=cal(now,0);
    get_size(now,0);
    vis[now]=1;
    for (int i=ls[now];i;i=e[i].next) {
        if (vis[e[i].y]) continue;
        ans-=cal(e[i].y,e[i].w);
        root=0; sum=size[e[i].y];
        get_root(e[i].y,0);
        solve(root);
    }
}
int main(void) {
    n=read();
    rep(i,2,n) {
        int x=read();
        int y=read();
        int w=read();
        addEdge(x,y,w);
    }
    sum=n;
    mx_son[root=0]=INF;
    get_root(1,0);
    solve(root);
    int GCD=gcd(n*n,ans);
    printf("%d/%d\n",ans/GCD,n*n/GCD);
    return 0;
}