BZOJ 2152: 聪聪可可 点分治

脑子不太清醒，看点分治很久才有点感觉，于是顶着睡意卡了一道题

点分治，算是入个门吧

学习点分治http://www.cnblogs.com/chty/p/5912360.html

相当于照着抄的

#include<cstdio>#include<algorithm>#define N 20005using namespace std;inline int read(){int a=0,f=1;static char c=getchar();while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}while(c>='0'&&c<='9'){a=a*10+c-'0';c=getchar();}return a*f;}struct edge{int to,v,next;}e[N*2];int n,sum,ans=0,root=0,cnt=0;int t[5],d[N],son[N],mx[N],vis[N],head[N];int gcd(int a,int b){if(b==0) return a;else return gcd(b,a%b);}void add(int x,int y,int v){e[++cnt].next=head[x];head[x]=cnt;e[cnt].to=y,e[cnt].v=v;}void getr(int x,int fa){son[x]=1,mx[x]=0;for(int i=head[x];i;i=e[i].next){if(e[i].to==fa||vis[e[i].to]) continue;getr(e[i].to,x);son[x]+=son[e[i].to];mx[x]=max(mx[x],son[e[i].to]);}mx[x]=max(mx[x],sum-son[x]);if(mx[x]<mx[root]) root=x;}void getdeep(int x,int fa){++t[d[x]];for(int i=head[x];i;i=e[i].next){if(e[i].to==fa||vis[e[i].to]) continue;d[e[i].to]=(d[x]+e[i].v)%3;getdeep(e[i].to,x);}}int cal(int x,int v){t[0]=t[1]=t[2]=0;d[x]=v;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){if(vis[e[i].to]) continue;ans-=cal(e[i].to,e[i].v);root=0,sum=son[e[i].to];getr(e[i].to,0);work(root);}}int main(){n=read();register int i;for(i=1;i<n;++i){int x=read(),y=read(),z=read()%3;add(x,y,z);add(y,x,z);}sum=mx[0]=n;getr(1,0);work(root);int t=gcd(ans,n*n);printf("%d/%d\n",ans/t,n*n/t);return 0;}