2152: 聪聪可可|树形DP

直接树形DP搞一搞，也可以点分治，然而有 O(n) 复杂度的树形DP为什么还要点分治呢QAQ
dis[x][i] 表示在以 x 为根的子树中，到 x 的距离为 i 的点的个数
发现距离直接对 3 取膜就可以，然后随便统计一下

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<iostream>
#include<algorithm>
#define N 400022
#define M 1000002
using namespace std;
int sc()
{
    int i=0,f=1; char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i*f;
}
int head[N],nxt[N],lst[N],v[N];
int n,dis[N][3],sum,tot;
int gcd(int x,int y){return x==0?y:gcd(y%x,x);}
void insert(int x,int y,int z)
{
    lst[++tot]=y;nxt[tot]=head[x];head[x]=tot;v[tot]=z;
    lst[++tot]=x;nxt[tot]=head[y];head[y]=tot;v[tot]=z;
}
void dfs(int x,int f)
{
    dis[x][0]=1;
    for(int i=head[x];i;i=nxt[i])
        if(lst[i]!=f)
        {
            dfs(lst[i],x);
            for(int j=0;j<3;j++)
                dis[x][(j+v[i])%3]+=dis[lst[i]][j];
        }
    sum+=dis[x][0];
    for(int i=head[x];i;i=nxt[i])
        if(lst[i]!=f)
            for(int j=0;j<3;j++)
                sum+=(dis[x][j]-dis[lst[i]][(j-v[i]+6)%3])*dis[lst[i]][(9-j-v[i])%3];
}
int main()
{
    n=sc();
    for(int i=1;i<n;i++)
    {
        int x=sc(),y=sc(),z=sc()%3;
        insert(x,y,z);
    }
    dfs(1,0);
    int ans=n*n,g=gcd(sum,ans);
    printf("%d/%d",sum/g,ans/g);
    return 0;
}