Codeforces 791D Bear and Tree Jump（树形DP）

题目链接 Bear and Tree Jumps

考虑树形DP。$c(i, j)$表示$i$最少加上多少后能被$j$整除。

在这里我们要算出所有$c(i, k)$的和。

其中$i$代表每个点对的距离，$k$为输入的$k$值。

$f[i][j]$表示以$i$为根结点，深度对$k$取模为$j$的点的个数。

状态转移时$f[x][i]$一边更新一边和刚刚计算出的$f[u][j]$统计答案。

具体细节可以看代码。

 1 #include <bits/stdc++.h>
 2 
 3 using namespace std;
 4 
 5 #define rep(i, a, b)    for (int i(a); i <= (b); ++i)
 6 
 7 const int N = 200010;
 8 int n, k;
 9 long long sum[N], f[N][6], ans = 0;
10 vector <int> v[N];
11 
12 void dfs(int x, int fa, int dep){
13     f[x][dep % k] = sum[x] = 1; //初始化
14     for (auto u : v[x]){
15         if (u == fa) continue;
16         dfs(u, x, dep + 1);
17         rep(i, 0, k - 1) rep(j, 0, k - 1){
18             int dis = ((i + j) % k - ((dep * 2) % k) + k) % k;
19             int t = (2 * k - dis) % k;
20             ans += t * f[x][i] * f[u][j]; //这一步求出要被k整除则还需补多少的总和 （1）
21         }
22 
23         rep(i, 0, k - 1) f[x][i] += f[u][i];
24         sum[x] += sum[u];
25         ans += (n - sum[u]) * sum[u]; //若没有（1）则这一步求的是树上所有点两两距离和 （2）
26     }
27 }
28 
29 int main(){
30 
31     scanf("%d%d", &n, &k);
32     rep(i, 1, n - 1){
33         int x, y;
34         scanf("%d%d", &x, &y);
35         v[x].push_back(y);
36         v[y].push_back(x);
37     }
38 
39     dfs(1, 0, 0);
40     printf("%lld\n", ans / k); //最后除以k
41     return 0;
42 }

 

转载于:https://www.cnblogs.com/cxhscst2/p/6696786.html