AC日记——【模板】点分治（聪聪可可） 洛谷 P2634

【模板】点分治（聪聪可可）

思路：

　　点分治；

　　（感谢灯神）

代码：

#include <bits/stdc++.h>
using namespace std;
#define maxn 20005
#define INF 0x7fffffff
int n,m,sum,num,cnt,ans,L,root,t;
int head[maxn],vis[maxn],d[maxn];
int size[maxn],lar[maxn],flag[];
int E[maxn<<],V[maxn<<],W[maxn<<];
inline void in(int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'')Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}
inline void edge_add(int u,int v,int w)
{
    E[++cnt]=head[u],V[cnt]=v,W[cnt]=w,head[u]=cnt;
    E[++cnt]=head[v],V[cnt]=u,W[cnt]=w,head[v]=cnt;
}
inline int gcd(int a,int b)
{
    return b==?a:gcd(b,a%b);
}
void GetRoot(int now,int fa)
{
    size[now]=,lar[now]=;
    for(int i=head[now];i;i=E[i])
    {
        if(V[i]==fa||vis[V[i]]) continue;
        GetRoot(V[i],now);
        size[now]+=size[V[i]];
        lar[now]=max(lar[now],size[V[i]]);
    }
    lar[now]=max(lar[now],sum-size[now]);
    if(lar[now]<lar[root]) root=now;
}
void GetDeep(int now,int fa)
{
    flag[d[now]]++;
    for(int i=head[now];i;i=E[i])
    {
        if(vis[V[i]]||V[i]==fa)continue;
        d[V[i]]=(d[now]+W[i])%;
        GetDeep(V[i],now);
    }
}
int cal(int now,int dis)
{
    d[now]=dis,flag[]=flag[]=flag[]=;
    GetDeep(now,);
    return flag[]*flag[]*+flag[]*flag[];
}
void work(int now)
{
    ans+=cal(now,),vis[now]=;
    for(int i=head[now];i;i=E[i])
    {
        if(vis[V[i]]) continue;
        ans-=cal(V[i],W[i]);
        root=,sum=size[V[i]];
        GetRoot(V[i],),work(root);
    }
}
int main()
{
    in(n);int u,v,w;
    for(int i=;i<n;i++)
    {
        in(u),in(v),in(w);
        edge_add(u,v,w%);
    }
    sum=n,lar[]=n+;
    GetRoot(,),work(root);
    t=gcd(ans,n*n);
    printf("%d/%d\n",ans/t,n*n/t);
    return ;
}