[树形dp + 组合数学]蓝魔法师

it2024-04-08  47

题目:蓝魔法师

分析:

dp[i][j]表示对于i这个子树来说,i所在的连通块大小为j,并且整棵树合法的方案数。 接下来就是类似背包的解法了。遍历每个儿子的分支,每个分支要么删除,要么不删除。 1.删除:

删除此边,那么就意味着当前以u节点连通块大小为k的方案数 都可以 乘 v节点连通块大小所有的方案数:

2.不删除

不删除就相当于合并那么此时直接跑两个循环即可:

for(int i=1;i<=min(sz[u]*1ll,m);i++){ for(int k=1;k<=min(sz[e]*1ll,m);k++){ if(i+k<=m){ tmp[i+k] += (dp[u][i]*dp[e][k])%mod; tmp[i+k] %=mod; } } }

考虑到和dp[u][k]的状态有关,所以用tmp数组先预初储存一下!

代码:

#include <bits/stdc++.h> using namespace std; typedef long long ll;//三年竞赛一场空,不开long long见祖宗 //typedef __int128 lll; #define print(i) cout << "debug: " << i << endl #define close() ios::sync_with_stdio(0), cin.tie(0), cout.tie(0) #define mem(a, b) memset(a, b, sizeof(a)) #define pb(a) push_back(a) #define x first #define y second typedef pair<int, int> par; const ll mod = 998244353; const int maxn = 2e3 + 10; const int inf = 0x3f3f3f3f; ll n, k; vector<int> g[maxn]; ll dp[maxn][maxn], tmp[maxn]; ll siz[maxn]; void dfs(int u, int fa) { siz[u] = 1, dp[u][1] = 1; for(int v : g[u]) { if(v == fa) continue; dfs(v, u); ll sum = 0; for(int i = 1; i <= min(siz[v], k); i++) sum += dp[v][i], sum %= mod; for(int i = 1; i <= min(siz[u], k); i++) for(int j = 1; j <= min(siz[v], k); j++) if(i + j <= k) tmp[i + j] += dp[u][i] * dp[v][j] % mod, tmp[i + j] %= mod; siz[u] += siz[v]; for(int i = 1; i <= min(siz[u], k); i++) dp[u][i] = (sum * dp[u][i] % mod + tmp[i]) % mod; mem(tmp, 0); } } int main() { cin >> n >> k; for(int i = 1; i < n; i++) { int a, b; cin >> a >> b; g[a].pb(b), g[b].pb(a); } dfs(1, -1); ll res = 0; for(int i = 1; i <= k; i++) res = (res + dp[1][i]) % mod; cout << res << endl; }
最新回复(0)