题目
题解
话说今晚随便水了水奶牛题,发现有的题虽然不难,但是质量确实高。以后没事就水水。
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;
}