题目:蓝魔法师
分析:
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;
}