洛谷 P1472 奶牛家谱 Cow Pedigrees

题目

传送门

题解

话说今晚随便水了水奶牛题，发现有的题虽然不难，但是质量确实高。以后没事就水水。

---

dp（i，j）表示i个节点深度为j的树有多少个。那么如果直接转移的话，需要先枚举左子树节点个数k，和一个dep（因为至少有一个子树的深度为j-1）然后如下转移。

dp（i，j）=sigma（dp（k，j-1）* dp（i-k-1，dep））+s igma（dp（k，dep）* dp（i-k-1，j-1））

注意：根据dep的枚举方式会有算重的或者算漏的要处理一下，上面只是个大概思路

复杂度明显过高。引入新数组 f（i，j）表示i个节点组成的深度不大于j的个数，那么方程大幅简化，分三种情况，因为至少有一个子树的深度为j-1，所以要么左子树深度为j-1，要么右子树是，要么都是。

dp（i，j）=dp（k，j-1）* f（i-k-1，j-2）+f（k，j-2）* dp（i-k-1，j-1） + dp（k，j-1） * dp（i-k-1，j-1）

然后可以看见只需要枚举k，复杂度n* n *k

---

其实一开始想的是递推dp时不分情况直接用f（k，j-1）*f（k-i-1，j-1）算，后来发现这样算出来的树深度不一定为j，所以就得分情况啦，不过还是挺水的。

代码

//QWsin
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int MOD=9901;
const int maxn=200+10;
const int maxk=100+10;

ll dp[maxn][maxk],f[maxn][maxk];

int main()
{
    int n,K;cin>>n>>K;
//  实际上数据没有n为奇数的 
//  if(n%2==0){printf("0\n");return 0;} 
    dp[1][1]=1;f[0][0]=1;
    for(int i=1;i<=K;i++) f[1][i]=f[0][i]=1;
    for(int i=3;i<=n;i+=2)//共n个节点 
    {
        int j=1;while((1<<j)-1 < i)j++;
        for(;j<=K;j++) //深度 
        {
            for(int k=1;k<=i-1;k+=2)//枚举左子树节点个数 
            {
                dp[i][j]+=dp[k][j-1]*f[i-k-1][j-2];
                dp[i][j]+=f[k][j-2]*dp[i-k-1][j-1];
                dp[i][j]+=dp[k][j-1]*dp[i-k-1][j-1];
            }
            dp[i][j]%=MOD;  
        }
        for(int j=1;j<=K;j++) f[i][j]=(f[i][j-1]+dp[i][j])%MOD;
    }
    printf("%lld\n",dp[n][K]);
    return 0;
}