题解 子树问题 dp

题解 子树问题

题目描述

数据范围:$n,k\le 500$

具体做法与心路历程

考试时想了一下，只能设出$dp$方程，然后不会转移。

具体做法

设$f[size][dep]$表示有$size$个节点，最大深度不超过$dep$的合法的树的数目。

考虑怎么转移$f[size][dep]$:

首先有$2$的父亲一定是$1$,那么我们枚举以$2$为根的大小i,那么除去这$i$个点其余点构成一颗以$1$为根，点数为$size-i$,最大深度为$dep$的树。选取$2$的子树中的点有$C_{size-2}^{i-1}$种方案。所以转移为:
$$f[size][dep]=\sum _{i=1}^{size-1}f[size-i][dep]\times f[i][dep-1]\times {(}_{j-1}^{i-2})$$
对于题目的限制条件我们把方案看成$0$即可。

$\mathcal{C}ode$

/*******************************
Author:galaxy yr
LANG:C++
Created Time:2019年11月02日 星期六 15时31分15秒
*******************************/
#include<cstdio>
#include<algorithm>

using namespace std;

struct IO{
    template<typename T>
    IO & operator>>(T&res)
    {
        T q=1;char ch;
        while((ch=getchar())<'0' or ch>'9')if(ch=='-')q=-q;
        res=(ch^48);
        while((ch=getchar())>='0' and ch<='9') res=(res<<1)+(res<<3)+(ch^48);
        res*=q;
        return *this;
    }
}cin;

const int mod=998244353;
const int maxn=505;
long long n,k,fac[maxn],ifac[maxn],f[maxn][maxn],l,r;
bool unable[505],vis[maxn][maxn];

void init()
{
    fac[0]=fac[1]=ifac[0]=ifac[1]=1;
    for(int i=2;i<=n;i++)
        fac[i]=1ll*fac[i-1]*i%mod,ifac[i]=1ll*(mod-mod/i)*ifac[mod%i]%mod;
    for(int i=2;i<=n;i++)
        ifac[i]=1ll*ifac[i-1]*ifac[i]%mod;
}

int C(int n,int m)
{
    return 1ll*fac[n]*ifac[m]%mod*ifac[n-m]%mod;
}

long long dp(int size,int dep)
{
    if(size<=0 || dep<=0) return 0;
    if(vis[size][dep]) return f[size][dep];
    vis[size][dep]=1;
    if(size==1) return f[size][dep]=1;
    if(!unable[size-1])
        f[size][dep]=dp(size-1,dep-1);//1只有2这个子树
    for(int i=1;i<=size-2;i++)
        if(!unable[i])
        {
            int res=1ll*C(size-2,i-1)*dp(i,dep-1)%mod*dp(size-i,dep)%mod;
            f[size][dep]=(f[size][dep]+res)%mod;
        }
    return f[size][dep];
}

int main()
{
    //freopen("subtree.in","r",stdin);
    //freopen("subtree.out","w",stdout);
    cin>>n>>k;
    init();
    for(int i=1;i<=k;i++)
    {
        int x;
        cin>>x;
        unable[x]=1;
    }
    cin>>l>>r;
    for(int i=l;i<=r;i++)
    {
        if(!unable[n])
            printf("%lld ",(dp(n,i)-dp(n,i-1)+mod)%mod);
        else
            printf("0 ");
    }
    putchar('\n');
    return 0;
}