点分治练习——BZOJ 2152

做的第二道点分治的题目，比较裸，算是模板题吧（感觉比之前那题还简单点。

题目:BZOJ 2152 聪聪可可

题目大意：给出一棵树，求树上两点间长度为3的倍数(0也算)的路径数。

解题思路：

基本和POJ1741一样

2.不过重心，在重心的子树中

情况二可通过分治转化为情况1。

通过dfs求出每个点到重心的距离%3，将余数是1的和是2的配对，余数是0的两两配对，得出路径数。

同样的，注意去重。

代码:

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 using namespace std;

 struct edge{
     int w,to,next;
 }e[];

 int head[],s[],f[],d[],t[],ans,size;
 bool b[];
 int n,i,j,v,w,x,y,ne=,root;

 void add(int a,int b,int c){
     e[++ne].to=b; e[ne].w=c; e[ne].next=head[a]; head[a]=ne;
 }

 void getroot(int k,int fa){
     int v,i;
     s[k]=;f[k]=;
     for(i=head[k];i!=-;i=e[i].next){
         v=e[i].to;
         if(v!=fa&&!b[v]){
             getroot(v,k);
             s[k]+=s[v];
             f[k]=max(f[k],s[v]);
         }
     }
     f[k]=max(f[k],size-s[k]);
     if(f[k]<f[root])root=k;
 }

 void getdis(int k,int fa){
     int v,i;
     t[d[k]]++;//将对应余数的数目+1；
     for(i=head[k];i!=-;i=e[i].next){
         v=e[i].to;
         if (v!=fa&&!b[v]){
             d[v]=(d[k]+e[i].w)%;
             getdis(v,k);
         }
     }
 }

 int clac(int k,int init){
     t[]=t[]=t[]=;//余数清0
     d[k]=init%;
     getdis(k,);//计算深度
     return (t[]*t[]*+t[]*t[]);//计算路径数
 }

 void work(int k){
     int i,v;
     ans+=clac(k,);//更新ans
     b[k]=true;
     for(i=head[k];i!=-;i=e[i].next){
         v=e[i].to;
         if(!b[v]){
             ans-=clac(v,e[i].w);//去重
             f[]=size=s[v];
             root=;
             getroot(v,);//更新重心
             work(root);//求解子树
         }
     }
 }

 int gcd(int a,int b){
     if(b==) return a;
     return gcd(b,a%b);
 }

 int main(){
     ans=root=;
     memset(head,-,sizeof(head));
     memset(b,,sizeof(b));
     scanf("%d",&n);
     for(i=;i<n;i++){
         scanf("%d%d%d",&x,&y,&w);
         w%=;
         add(x,y,w);
         add(y,x,w);//连双向边
     }
     f[]=size=n;
     getroot(,);//求重心
     work(root);
     int tmp=gcd(ans,n*n);
     printf("%d/%d\n",ans/tmp,n*n/tmp);
 }