bzoj 2152: 聪聪可可

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=20010;
int t[5],head[N],son[N],f[N],d[N],root,ans,n,sum,cnt;
bool vis[N];
struct ee{int to,next,v;}e[N*2];
void insert(int u,int v,int w){
    e[++cnt].to=v;e[cnt].next=head[u];e[cnt].v=w;head[u]=cnt;
}
 
int gcd(int a1,int a2)
{
    int a3=a1%a2;
    for(;a3;)
      {
        a1=a2;
        a2=a3;
        a3=a1%a2;
      }
    return a2;
}
 
void getroot(int x,int fa){
    son[x]=1;f[x]=0;
    for (int i=head[x];i;i=e[i].next){
        int v=e[i].to;
        if (!vis[v]&&v!=fa){
            getroot(v,x);
            son[x]+=son[v];
            f[x]=max(f[x],son[v]);
        }
    }
    f[x]=max(f[x],sum-son[x]);
    if (f[root]>f[x])root=x;
}
 
void getdeep(int x,int fa){
    t[d[x]]++;
    for (int i=head[x];i;i=e[i].next){
        int v=e[i].to;
        if (v!=fa&&!vis[v]){
            d[v]=(d[x]+e[i].v)%3;
            getdeep(v,x);
        }
    }
}
 
int cal(int x,int now){
    t[0]=t[1]=t[2]=0;
    d[x]=now;
    getdeep(x,0);
    return t[1]*t[2]*2+t[0]*t[0];
}
 
void work(int x){
    ans+=cal(x,0);
    vis[x]=1;
    for (int i=head[x];i;i=e[i].next){
        int v=e[i].to;
        if (!vis[v]) {
            ans-=cal(v,e[i].v);
            root=0;sum=son[v];
            getroot(v,0);
            work(root);
        }
    }
}
 
int main(){
    scanf("%d",&n);
    int u,v,w;
    for (int i=1;i<n;i++){
        scanf("%d%d%d",&u,&v,&w);
        w%=3;
        insert(u,v,w);
        insert(v,u,w);
    }
    f[0]=sum=n;
    root=0;
    getroot(1,0);
    work(root);
    int t=gcd(ans,n*n);
    printf("%d/%d",ans/t,n*n/t);
}

树的点分治  首先找树的重心，把根节点设为树的重心。然后进行点分治(就是从根开始，看经过根的方案数，在把根删掉，求每棵子树)。