当前位置: 代码迷 >> 综合 >> 【CodeForces - 161D 】【1800】【树形DP】
  详细解决方案

【CodeForces - 161D 】【1800】【树形DP】

热度:66   发布时间:2024-01-04 11:36:13.0

https://codeforces.com/problemset/problem/161/D

【题意】

n个节点的树,边权为1,问你长度为k的对数有多少对

【思路】

树形dp

dp[x][i]+=dp[y][i-1];(x--y是有一条边)

每次转移更新dp前,记录答案贡献

ans=dp[x][i]*dp[y][k-i-1];(表示这个子树宇之前所有的组合对数)

【代码】

#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e4 + 5;vector<int>vec[maxn];
int dp[maxn][506];
int ans = 0;
int n, k;void dfs(int x, int fa) {dp[x][0] = 1;for (int y : vec[x]) {if (y == fa)continue;dfs(y, x);for (int i = 0; i < k; i++) {ans += dp[x][i] * dp[y][k - i - 1];}for (int i = 1; i <= k; i++) {dp[x][i] += dp[y][i - 1];}}
}int main() {scanf("%d%d", &n, &k);for (int i = 1; i < n; i++) {int x, y;scanf("%d%d", &x, &y);vec[x].push_back(y);vec[y].push_back(x);}dfs(1, -1);printf("%d\n", ans);
}