bzoj2152 聪聪可可（点分治/树形dp）

就是求路径和%3余0的有序点对个数。可以点分治。还可以树形dp。
dp[i][j]表示以i为根的子树中的点，到i的路径长度%3余j的个数。每次统计一下过i的答案就好了。

点分治

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define N 20010
#define inf 0x3f3f3f3f
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,h[N],num=0,rt,sz[N],sumsz,ans=0,fa[N],cnt[3],f[N],d[N];
bool vis[N];
struct edge{
    int to,next,val;
}data[N<<1];
inline void dfs1(int x){
    sz[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]||y==fa[x]) continue;
        fa[y]=x;dfs1(y);sz[x]+=sz[y];
    }
}
inline void dfs2(int x){
    f[x]=0;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]||vis[y]) continue;
        dfs2(y);f[x]=max(f[x],sz[y]);
    }f[x]=max(f[x],sumsz-sz[x]);if(f[x]<f[rt]) rt=x;
}
inline void dfs3(int x){
    cnt[d[x]]++;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]||vis[y]) continue;
        d[y]=(d[x]+data[i].val)%3;dfs3(y);
    }
}
inline int calc(int x,int val){
    d[x]=val;memset(cnt,0,sizeof(cnt));dfs3(x);
    return cnt[1]*cnt[2]*2+cnt[0]*cnt[0];
}
inline void solve(int x){
    vis[x]=1;fa[x]=0;dfs1(x);ans+=calc(x,0);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]) continue;
        ans-=calc(y,data[i].val);rt=0;sumsz=sz[y];dfs2(y);solve(rt);
    }
}
inline int gcd(int x,int y){return y==0?x:gcd(y,x%y);}
int main(){
//  freopen("a.in","r",stdin);
    n=read();f[0]=inf;
    for(int i=1;i<n;++i){
        int x=read(),y=read(),w=read()%3;
        data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=w;
        data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=w;
    }dfs1(1);rt=0;sumsz=n;dfs2(1);solve(rt);
    int tot=n*n,gg=gcd(ans,tot);
    printf("%d/%d",ans/gg,tot/gg);
    return 0;
}

树形dp

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define N 20010
#define inf 0x3f3f3f3f
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,h[N],num=0,ans=0,fa[N],dp[N][3];
inline int gcd(int x,int y){return y==0?x:gcd(y,x%y);}
struct edge{
    int to,next,val;
}data[N<<1];
inline void dfs(int x){
    dp[x][0]=1;ans++;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]) continue;
        fa[y]=x;dfs(y);
        for(int j=0;j<3;++j){
            int k=(j+data[i].val)%3;k=3-k;if(k==3) k=0;
            ans+=dp[y][j]*dp[x][k]*2;
        }for(int j=0;j<3;++j){
            int k=(j+data[i].val)%3;
            dp[x][k]+=dp[y][j];
        }
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read(),w=read()%3;
        data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=w;
        data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=w;
    }dfs(1);
    int tot=n*n,gg=gcd(ans,tot);
    printf("%d/%d",ans/gg,tot/gg);
    return 0;
}