蓝魔法师

链接：https://ac.nowcoder.com/acm/contest/6895/A
来源：牛客网
 

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

“你，你认错人了。我真的，真的不是食人魔。”--蓝魔法师

 

给出一棵树，求有多少种删边方案，使得删后的图每个连通块大小小于等于k，两种方案不同当且仅当存在一条边在一个方案中被删除，而在另一个方案中未被删除，答案对998244353取模

输入描述:

第一行两个整数n,k, 表示点数和限制
2 <= n <= 2000, 1 <= k <= 2000
接下来n-1行，每行包括两个整数u,v，表示u,v两点之间有一条无向边
保证初始图联通且合法

输出描述:

共一行，一个整数表示方案数对998244353取模的结果

示例1

输入

5 2
1 2
1 3
2 4
2 5

输出

7

思路：树形dp,dp[i][j]表示以i为根节点的大小为j的连通图有多少个。如果u与v相连，那么dp[u][i]*dp[v][j]即为大小为i+j的连通图大小。

#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<cstring>

using namespace std;

const int N = 2020;
const int mod = 998244353;
typedef long long ll;

int n,k;
vector<int> g[N];
int siz[N];
ll dp[N][N];
ll tem[N];

void dfs(int u, int fa)
{
	dp[u][1] = siz[u] = 1;
	
	for(int i=0; i<g[u].size(); i++)
	{
		int v = g[u][i];
		if(v == fa) continue;
		dfs(v,u);
		
		memset(tem, 0, sizeof tem);
		for(int j = 1; j <= siz[u]; j++)
			for(int f = 0; f <= siz[v] && f + j <= k; f++)
				tem[j+f] = (tem[j+f] + dp[u][j] * dp[v][f] % mod) % mod;
		for(int j=1; j<=k; j++)
			dp[u][j] = tem[j];
		siz[u] += siz[v];
	}
	
	for(int i=1; i<=k; i++) dp[u][0] = (dp[u][0] + dp[u][i]) % mod;
}

int main()
{
	cin>>n>>k;
	for(int i=0; i<n-1; i++)
	{
		int u,v;
		cin>>u>>v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	dfs(1,0);
	cout<<dp[1][0];
	return 0;
}