P9414 「NnOI R1-T3」元组 （树上背包）

P9414 「NnOI R1-T3」元组

树上背包

首先思考题意，每个方案都存在一个唯一的 $x$，所以我们可以枚举 $x$，计算有多少方案使得 $\mathrm{L}\mathrm{C}\mathrm{A}$ 为 $x$。

$x$ 上方的点一定不能选，那么就变成了在 $x$ 子树内的选点问题。思考后可以发现，要满足题意，就是要满足 每个 $so{n}_u$ 子树中选的点 $<k$！那么这就是一个裸的树上背包问题，设 $f_{u,i}$ 表示在 $u$ 中选 $i$ 个点的方案数，调整一下枚举上界即可。

复杂度 $O(n^2)$。

我竟然能做出来，难度虚高

#include <bits/stdc++.h>
#define pii std::pair<int, int>
#define fi first
#define se second
#define pb push_back

typedef long long i64;
const i64 iinf = 0x3f3f3f3f, linf = 0x3f3f3f3f3f3f3f3f;
const int N = 5e3 + 10, mod = 1e9 + 7;
i64 n, p, k;
i64 f[N][N], sz[N], ans, tmp[N];
std::vector<int> e[N];
void dfs(int u, int fa) {
	sz[u] = 1;
	f[u][0] = f[u][1] = 1;
	for(auto v : e[u]) {
		if(v == fa) continue;
		dfs(v, u);
		for(int i = 0; i <= sz[u] + sz[v]; i++) tmp[i] = 0;
		for(int i = 0; i <= sz[u]; i++) {
			for(int j = 0; j <= std::min(k - 1, sz[v]); j++) {
				tmp[i + j] = (tmp[i + j] + f[u][i] * f[v][j]) % mod;
			}
		}
		for(int i = 1; i <= sz[u] + sz[v]; i++) f[u][i] = tmp[i];
		sz[u] += sz[v];
	}
	ans = (ans + f[u][p]) % mod;
}
void Solve() {
	std::cin >> n >> p >> k;
	for(int i = 1; i < n; i++) {
		int u, v;
		std::cin >> u >> v;
		e[u].pb(v), e[v].pb(u);
	}
	dfs(1, 0);
	std::cout << ans << "\n";
}
int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    
	Solve();

	return 0;
}