I-2019 换根dp

I-2019
题意：求从u到v所经过的权值相加%2019==0的对数，要求u>v。
题解：首先你要知道怎么求固定根，用树形dp写出，可以仿照这类题的解法，设dp[i][j]代表节点i权值为j%2019的点的个数,然后换根dp。这个换根dp跟我之前写的不完全一样，建议看代码学习一下。

#include<bits/stdc++.h>
#define pi pair<int,int>
#define mk make_pair
#define ll long long
using namespace std;
const int maxn = 2e4+10,mod = 2019;
vector<pi>G[maxn];
int dp[maxn][2020];
ll ans = 0;
void dfs(int u,int fa)
{
    for(auto tmp :G[u])
    {
        int v = tmp.first , w = tmp.second;
        if(v==fa) continue;
        dfs(v,u);
        dp[u][w]++;
        for(int i=0;i<2019;i++)
        dp[u][(i+w)%mod] += dp[v][i];
    }
}
void dfs2(int u,int fa){
     for(auto tmp :G[u])
     {
         int v = tmp.first , w = tmp.second;
         if(v==fa) continue;
         int d1[2019],d2[2019];
         for(int i=0;i<2019;i++)
         d1[i] = dp[u][i],d2[i]=dp[v][i];
         for(int i=0;i<2019;i++)
         dp[u][(i+w)%mod] -= dp[v][i];
         dp[u][w]--;
         dp[v][w]++;
         for(int i=0;i<2019;i++)
        dp[v][(i+w)%mod] += dp[u][i];
        dfs2(v,u);
         ans += dp[v][0];
         for(int i=0;i<2019;i++)
         dp[u][i] = d1[i] , dp[v][i] = d2[i];
     }
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        int u,v,w;
        for(int i=1;i<=n;i++)G[i].clear();
        for(int i=1;i<=n;i++)
            for(int j=0;j<=2019;j++)
             dp[i][j] = 0;
        ans = 0;
        for(int i=1;i<n;i++)
        {
         scanf("%d%d%d",&u,&v,&w);
         G[u].push_back(mk(v,w));
         G[v].push_back(mk(u,w));
        }
        dfs(1,0);
        ans = dp[1][0];
        dfs2(1,0);
        printf("%lld\n",ans/2);
    }
}