P2634 [国家集训队]聪聪可可（树形dp）

这道题主要是要学习下这个树形dp的写法。
我一开始的想法是全部转移完再来统计贡献。然而，这就造成一个麻烦。
我们要统计的贡献是树中是权值和为3的倍数的路径个数。如果按照暴力思想，肯定是枚举两个点之后check这两点之间的距离是不是3的倍数。
然而，2e4的数据量不允许我们这样做。因此，考虑到这道题是树上问题，再联想到树上求距离还有一个重要的对象就是lca。所以，我们的状态设计其实就是：dp[u][i]表示到点u的距离模3为i的点的个数，相当于枚举了lca。
那么，考虑统计贡献：显然，贡献来自dp[u][0]以及dp[u][1]+dp[u][2]这两个相加起来等于3的路径。但是注意到，这个dp[u][1]和dp[u][2]必须来自不同的儿子，不然他们取得lca就不是u。
因此，我们不能全部转移完再来统计。考虑在转移途中统计贡献。

AC代码

#include <bits/stdc++.h>
using namespace std;
template<typename T>
void read(T&x){
    x=0;
    char ch=getchar();
    int f=1;
    while(!isdigit(ch)){
        if(ch=='-')f*=-1;
        ch=getchar();
    }
    while(isdigit(ch)){
        x=x*10+(ch-'0');
        ch=getchar();
    }x*=f;
}
template<typename T>
void write(T x){
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10);
    putchar(x%10+'0');
}
//=============================================
typedef long long ll;
#define int ll
const int maxn=2e4+10;
int n;
struct node{
    int to,next,w;
}e[maxn<<1];
int head[maxn],cnt;
void add(int x,int y,int w){
    e[cnt].to=y;
    e[cnt].w=w;
    e[cnt].next=head[x];
    head[x]=cnt++;
}
bool vis[maxn];
int dp[maxn][3];
int sum=0;
void solve(int u,int fa){
    dp[u][0]=1;
    for(int i=head[u];~i;i=e[i].next){
        int v=e[i].to;
        if(v==fa)continue;
        solve(v,u);
        for(int j=0;j<3;++j){
            sum+=2*dp[u][j]*dp[v][((3-e[i].w-j)%3+3)%3];
        }
        for(int j=0;j<3;++j){
            dp[u][(j+e[i].w)%3]+=dp[v][j];
        }
    }
}

signed main(){
    //freopen("in.txt","r",stdin);
    memset(head,-1,sizeof(head));
    read(n);
    for(int i=1;i<n;++i){
        int x,y,z;
        read(x),read(y),read(z);
        add(x,y,z);add(y,x,z);
    }
    int tot=n*n;
    solve(1,-1);
    sum+=n;
    int gg=__gcd(sum,tot);
    cout<<(sum/gg)<<"/"<<(tot/gg)<<endl;
    return 0;
}