bzoj2152 聪聪与可可 点分治

---

这一道题比poj1741那道入门题要难一点，不过也很水.点分治后我们在算dis的时候我们每次都要%3，因为mod可以分开加再mod，这是一个性质，所有与重心dis为1（%3=1）与2的可以刚好组成一组三的倍数的边，由题可知（4,5）和（5,4）是不同的一对，所以我们还要乘以2.注意本题自己到自己距离为零（%3=0）也算是一个点，所以我们算到中心距离刚好是三的倍数的点直接t[0]*t[0](t[0]）表示为三的倍数，自己到自己，自己到其他的同样到重心的距离是三的倍数的点的距离肯定都是三的倍数.
注意本题要输出最简分数，要gcd；

#include<stdio.h>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=20005;
bool vis[maxn];
int dis[maxn],h[maxn*3],f[maxn],root,n,siz[maxn],num,ans,sum,gc,t[4];
inline const int read(){
    register int f=1,x=0;
    register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}
struct edge{
    int v,nxt,w;
}e[maxn*3];
void add(int u,int v,int w){
    e[++num].v=v;
    e[num].w=w;
    e[num].nxt=h[u];
    h[u]=num;
}
void getroot(int u,int fa){
    siz[u]=1,f[u]=0;
    for(int i=h[u];i;i=e[i].nxt){
        int v=e[i].v;
        if(v==fa||vis[v]) continue;
        getroot(v,u);
        siz[u]+=siz[v];
        f[u]=max(f[u],siz[v]);
    }
    f[u]=max(f[u],sum-siz[u]);
    if(f[root]>f[u]) root=u;;
}
void getdeep(int u,int fa){
    t[dis[u]]++;
    for(int i=h[u];i;i=e[i].nxt){
        int v=e[i].v;
        if(v==fa||vis[v]) continue;
        dis[v]=(dis[u]+e[i].w)%3;
        getdeep(v,u); 
    }
} 
inline int cal(int u,int now){
    t[0]=t[1]=t[2]=0;
    dis[u]=now;
    getdeep(u,0);
    return t[1]*t[2]*2+t[0]*t[0];
}
void solve(int u){printf("%d\n",root);
    ans+=cal(u,0);
    vis[u]=true;
    for(int i=h[u];i;i=e[i].nxt){
        int v=e[i].v;
        if(vis[v]) continue;
        ans-=cal(v,e[i].w);
        sum=siz[v];
        root=0;
        getroot(v,root);
        solve(root);
    }
}
int gcd(int a,int b){
    return b==0?a:gcd(b,a%b);
}
int main(){
    n=read();
    int x,y,w;
    for(register int i=1;i<=n-1;i++)
     x=read(),y=read(),w=read(),w=w%3,add(x,y,w),add(y,x,w);
    f[0]=sum=n;
    getroot(1,0);    
    solve(root);
    int gc=gcd(ans,n*n);
    printf("%d/%d\n",ans/gc,n*n/gc);
    return 0;
}