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

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);
}