[bzoj2152]聪聪可可

树分治。

边的话存的时候存成mod 3后的边就行，然后其他的就是套模板了。（数组开太大会被T的。。。）

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
using namespace std;

const int N=50000;
int h[20010],r[40010],to[40010],w[40010],d[20010],sz[20010],f[20010],root,sum,tot,n,ans,t[5];
bool v[20010];
int gcd(int a,int b){return b?gcd(b,a%b):a;}
void ins(int u,int v,int z){
    to[tot]=v;
    w[tot]=z%3;
    r[tot]=h[u];
    h[u]=tot++;
}
void getRoot(int x,int fa){
    sz[x]=1;f[x]=0;
    for(int i=h[x];i!=-1;i=r[i]){
        if(!v[to[i]]&&to[i]!=fa){
            getRoot(to[i],x);
            sz[x]+=sz[to[i]];
            f[x]=max(f[x],sz[to[i]]);
        }
    }
    f[x]=max(f[x],sum-sz[x]);
    if(f[x]<f[root])root=x;
}
void getDeep(int x,int fa){
    t[d[x]]++;
    for(int i=h[x];i!=-1;i=r[i]){
        if(!v[to[i]]&&to[i]!=fa){
            d[to[i]]=(d[x]+w[i])%3;
            getDeep(to[i],x);
        }
    }
}
int calc(int x,int val){
    d[x]=val;t[0]=t[1]=t[2]=0;
    getDeep(x,0);
    return t[0]*t[0]+t[1]*t[2]*2;
}
void work(int x){
    ans+=calc(x,0);
    v[x]=1;
    for(int i=h[x];i!=-1;i=r[i]){
        if(!v[to[i]]){
            ans-=calc(to[i],w[i]);
            root=0;
            sum=sz[to[i]];
            getRoot(to[i],0);
            work(root);
        }
    }
}
inline int read(){
    static int f,ret;
    static char c;
    ret=0;f=1;
    c=getchar();
    while(c>'9'||c<'0'){
        if(c=='-')f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9')ret=(ret<<1)+(ret<<3)+c-'0',c=getchar();
    return ret*f;
}
int main(){
    n=read();
    memset(h,-1,sizeof(h));
    for(int i=1,u,v,z;i<n;i++){
        u=read();
        v=read();
        z=read();
        ins(u,v,z);ins(v,u,z);
    }
    f[0]=sum=n;
    getRoot(1,0);
    work(root);
    int g=gcd(ans,n*n);
    printf("%d/%d\n",ans/g,n*n/g);
}