51nod 1353 树

树背包

设f[i][j]表示第i个点，和子节点组成的联通块大小为j，其他都可行的方案

j=0表示可行的总方案

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
const int mod=1e9+;
int read()
{
    int x=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(''<=ch&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x;
}

int n,K;
struct node
{
    int x,y,next;
}a[];int len,last[];
void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
int f[][],tot[];
void dfs(int x,int fr)
{
    f[x][]=;tot[x]=;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fr)
        {
            dfs(y,x);
            for(int i=tot[x];i>=;i--)
            {
                for(int j=tot[y];j>=;j--)
                    f[x][i+j]=(f[x][i+j]+((LL)f[x][i]*(LL)f[y][j])%mod)%mod;
                f[x][i]=((LL)f[x][i]*(LL)f[y][])%mod;
            }
            tot[x]+=tot[y];
        }
    }
    for(int i=K;i<=tot[x];i++)f[x][]=(f[x][]+f[x][i])%mod;
}

int main()
{
    int x,y;
    n=read(),K=read();
    len=;memset(last,,sizeof(last));
    for(int i=;i<n;i++)
    {
        x=read(),y=read();
        ins(x,y),ins(y,x);
    }
    memset(f,,sizeof(f));
    dfs(,);
    printf("%d\n",f[][]);
    return ;
}