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+7;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while('0'<=ch&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x;
}

int n,K;
struct node
{
    int x,y,next;
}a[210000];int len,last[5100];
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[5100][5100],tot[5100];
void dfs(int x,int fr)
{
    f[x][1]=1;tot[x]=1;
    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>=1;i--)
            {
                for(int j=tot[y];j>=1;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][0])%mod;
            }
            tot[x]+=tot[y];
        }
    }
    for(int i=K;i<=tot[x];i++)f[x][0]=(f[x][0]+f[x][i])%mod;
}

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

 

转载于:https://www.cnblogs.com/AKCqhzdy/p/9806490.html