思路
思路应该算是比较简单吧...
难点在于时间复杂度的证明.
因为这题和
的范围都是
以内的.题目是让你切成若干份,每份大小都不超过
,因为是一棵树,所以很容易想到树形
.对于树形
,我们很容易想到利用子树就行转移.那么我们要转移什么,才能让子树
来跟新父节点
呢?很显然的一个东西,我
可以和子树分割,也可以不和子树分割.我们定义
的状态
表示到了
这个节点所连的联通块大小为
的方案数.
那么显然的转移方程就是:
假如不切割这颗子树,
.
假如切割这颗子树,那么就相当于一个独立事件,对于
的每种连通块大小都有可能,然后相乘即使答案了.
下面就是时间复杂度证明,简单一看好像是的.不过其实跑不满~~.时间真的卡的很紧,但绝对不是
的.
倒叙枚举,防止更新的答案作为结果更新后面的答案.
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=998244353;
const int N=2e3+5;
ll n,k;
vector<int>g[N];
ll f[N][N],sz[N];
void dfs(int u,int fa)
{
sz[u]=1;f[u][1]=1;
for(int v:g[u])
{
if(v==fa) continue;
dfs(v,u);
ll sum=0;
for(int i=1;i<=min(sz[v],k);i++) sum+=f[v][i],sum%=mod;
for(int i=min(sz[u],k);i>=1;i--)//枚举父亲节点的连通块的大小.
{
for(int j=min(sz[v],k);j>=1;j--)//枚举子节点的连通块大小.
if(i+j<=k) f[u][i+j]+=f[u][i]*f[v][j],f[u][i+j]%=mod;
f[u][i]*=sum,f[u][i]%=mod;
}
sz[u]+=sz[v];
}
}
int main()
{
scanf("%lld%lld",&n,&k);
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
g[x].push_back(y);
g[y].push_back(x);
}
dfs(1,1);
ll ans=0;
for(int i=1;i<=k;i++) ans+=f[1][i],ans%=mod;
printf("%lld\n",ans);
return 0;
}

京公网安备 11010502036488号