2019牛客国庆集训派对day1 I-2019

题目链接：https://ac.nowcoder.com/acm/contest/1099/I

解题心得：就是一个树$dp$，从叶子节点到根节点一步一步转移，用$dp[i][j]$表示以$i$为起点向叶子节点寻找路径能够组成路径和模2019为$j$的方案数有多少，这样在父节点找了一个子节点之后将子节点的方案数加到父节点上然后继续找下一个子节点。

---

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll maxn = 2e4+100;

int dp[maxn][2030], n;
ll ans = 0;
vector <pair<int, int> > ve[maxn];

void init() {
    ans = 0;
    for(int i=1;i<=n;i++)   ve[i].clear();
    for(int i=1;i<n;i++) {
        int a, b, c; scanf("%d%d%d",&a, &b, &c);
        ve[a].push_back({b, c});
        ve[b].push_back({a, c});
        memset(dp[i], 0, sizeof(dp[i]));
    }
    memset(dp[n], 0, sizeof(dp[n]));
    for(int i=1;i<=n;i++) {
        dp[i][0] = 1;
    }
}

void dfs(int pre, int now) {
    for(int i=0;i<ve[now].size();i++) {
        int v = ve[now][i].first;
        if(v == pre) continue;
        dfs(now, v);
    }
    for(int i=0;i<ve[now].size();i++) {
        int v = ve[now][i].first;
        int va = ve[now][i].second;
        if(v == pre) continue;
        for(int j=0;j<2019;j++) {
            ans += 1ll * dp[v][j] * dp[now][(2019-j-va+2019)%2019];
        }
        for(int j=0;j<2019;j++) {
            dp[now][(j+va)%2019] += dp[v][j];
        }
    }
}

int main() {
//    freopen("1.in.txt", "r", stdin);
    while(scanf("%d", &n) != EOF ){
        init();
        dfs(-1, 1);

        printf("%lld\n", ans);
    }
    return 0;
}