BZOJ2152——聪聪可可（点分治）

传送门

感觉点分治的题都差不多啊

基本上所有模板都是相似的

基本上只有$calc$一个版块有些区别

考虑到要边权和是3的倍数

所以就只会有1和2或者是3满足

所以我们dfs求出每个点到当前根节点的距离%3的值

并分别统计%后为0、1、2的个数为$num[0]、num[1]、num[2]$

最后答案就是$num[2]*num[1]*2+num[0]*num[0]$

然后除以总情况就是了

#include<bits/stdc++.h>
using namespace std;
#define ll long long
inline int read(){
    char ch=getchar();
    int res=0,f=1;
    while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)) res=(res<<3)+(res<<1)+(ch^48),ch=getchar();
    return res*f;
}
const int N=20005;
int n,adj[N],nxt[N<<1],to[N<<1],son[N],root,maxn,siz[N],val[N<<1],d[N],sum[4],cnt;
bool vis[N];
ll ans;
inline int gcd(int a,int b){
    return b==0?a:gcd(b,a%b);
}
inline void addedge(int u,int v,int w){
    nxt[++cnt]=adj[u],adj[u]=cnt,to[cnt]=v,val[cnt]=w;
    nxt[++cnt]=adj[v],adj[v]=cnt,to[cnt]=u,val[cnt]=w;
}
inline void getroot(int u,int fa){
    siz[u]=1,son[u]=0;
    for(int e=adj[u];e;e=nxt[e]){
        int v=to[e];
        if(v==fa||vis[v]) continue;
        getroot(v,u);
        siz[u]+=siz[v];
        son[u]=max(son[u],siz[v]);
    }
    son[u]=max(son[u],maxn-siz[u]);
    if(son[u]<son[root]) root=u;
}
inline void getdis(int u,int fa){
    sum[d[u]]++;
    for(int e=adj[u];e;e=nxt[e]){
        int v=to[e];
        if(vis[v]||v==fa) continue;
        d[v]=(d[u]+val[e])%3;
        getdis(v,u);
    }
}
inline int calc(int u,int lenth){
    sum[0]=sum[1]=sum[2]=0;
    d[u]=lenth;
    getdis(u,0);
    return sum[0]*sum[0]+sum[1]*sum[2]*2;
}
inline void solve(int u){
    ans+=calc(u,0);vis[u]=1;
    for(int e=adj[u];e;e=nxt[e]){
        int v=to[e];
        if(vis[v])continue;
        ans-=calc(v,val[e]) ;
        maxn=siz[v];
        getroot(v,root=0);
        solve(root);
    }
}
int main(){
    n=read();
    for(int i=1;i<n;i++){
        int u=read(),v=read(),w=read();
        w%=3;
        addedge(u,v,w);
    }
    maxn=n,son[0]=n;
    root=0,ans=0;
    getroot(1,0);
    solve(root);
    int a=gcd(ans,n*n);
    cout<<ans/a<<"/"<<n*n/a;
    return 0;
}