正題

題目鏈接:http://noip.ybtoj.com.cn/contest/90/problem/2

---

題目大意

$n$個點的一棵樹，求刪掉一些邊讓剩下的連通塊大小不小于$k$，求方案數。

---

解題思路

定義$f_{i,j}$表示$i$的子樹現在聯通塊大小為$j$時的方案數，我們不難發現每次$j$的上限就是$si{z}_x$，所以做一次就是$si{z}_x?si{z}_y$的復雜度，也就是每個點對直接貢獻一次。

時間復雜度$O(n^2)$

---

$code$

#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); }

總結

以上是生活随笔為你收集整理的YbtOJ#20064-[NOIP2020模拟赛B组Day4]预算缩减【树形dp】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。