正题
题目链接:http://noip.ybtoj.com.cn/contest/90/problem/2
题目大意
nnn个点的一棵树,求删掉一些边让剩下的连通块大小不小于kkk,求方案数。
解题思路
定义fi,jf_{i,j}fi,j表示iii的子树现在联通块大小为jjj时的方案数,我们不难发现每次jjj的上限就是sizxsiz_xsizx,所以做一次就是sizx∗sizysiz_x*siz_ysizx∗sizy的复杂度,也就是每个点对直接贡献一次。
时间复杂度O(n2)O(n^2)O(n2)
codecodecode
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=5100,XJQ=786433;
struct node{ll to,next;
}a[N*2];
ll n,k,tot,ans,ls[N],siz[N],f[N][N],g[N];
void addl(ll x,ll y){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;
}
void dfs(ll x,ll fa){siz[x]=1;f[x][1]=1;for(ll i=ls[x];i;i=a[i].next){ll y=a[i].to;if(y==fa)continue;dfs(y,x);for(ll i=1;i<=siz[x]+siz[y];i++)g[i]=0;for(ll i=1;i<=siz[x];i++){for(ll j=1;j<=siz[y];j++)(g[i+j]+=f[x][i]*f[y][j]%XJQ)%=XJQ;for(ll j=k;j<=siz[y];j++)(g[i]+=f[x][i]*f[y][j]%XJQ)%=XJQ;}siz[x]+=siz[y];for(ll i=1;i<=siz[x];i++)f[x][i]=g[i];}return;
}
int main()
{freopen("cut.in","r",stdin);freopen("cut.out","w",stdout);scanf("%lld%lld",&n,&k);for(ll i=1;i<n;i++){ll x,y;scanf("%lld%lld",&x,&y);addl(x,y);addl(y,x);}dfs(1,1);for(ll i=k;i<=siz[1];i++)(ans+=f[1][i])%=XJQ;printf("%lld",ans);
}