[点分治]BZOJ 2152——聪聪可可

题目描述

给出一棵有边权的树。

求边权和%3=0的路径数。

解题思路

累计以每个点为路径LCA的方案数，所以直接点分就好了。

累计方法为分别找出距离某个点路径和为0,1,2的方案数，返回$2*t[1]*t[2]+t[0]*t[0]$

然后减去所以子树方案就是当前方案。

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=20005;
int tot,lnk[maxn],nxt[2*maxn],son[2*maxn],w[2*maxn];
int n,ro,t[5],s[maxn],sum,F[maxn],ans,d[maxn];
bool vis[maxn];
inline int _read(){
    int num=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
    return num;
}
void add(int x,int y,int z){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;w[tot]=z;}
int gcd(int x,int y){if (!y) return x;return gcd(y,x%y);}
void getro(int x,int fa){
    s[x]=1;F[x]=0;
    for (int j=lnk[x];j;j=nxt[j]) if (!vis[son[j]]&&son[j]!=fa){
        getro(son[j],x);
        s[x]+=s[son[j]];
        F[x]=max(F[x],s[son[j]]);
    }
    F[x]=max(F[x],sum-s[x]);
    if (F[x]<F[ro]) ro=x;
}
void DFS(int x,int fa){
    t[d[x]]++;
    for (int j=lnk[x];j;j=nxt[j]) if (!vis[son[j]]&&son[j]!=fa){
        d[son[j]]=(d[x]+w[j])%3;
        DFS(son[j],x);
    }
}
int cal(int x,int v){
    t[0]=t[1]=t[2]=0;d[x]=v;
    DFS(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 j=lnk[x];j;j=nxt[j]) if (!vis[son[j]]){
        ans-=cal(son[j],w[j]%3);
        ro=0;sum=s[son[j]];
        getro(son[j],0);work(ro);
    }
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    n=_read();F[0]=n;
    for (int i=1;i<n;i++){
        int x=_read(),y=_read(),z=_read();
        add(x,y,z);add(y,x,z);
    }
    sum=n;getro(1,0);work(ro);
    int k=gcd(ans,n*n);
    printf("%d/%d\n",ans/k,n*n/k);
    return 0;
}