CS 300 BST Fixed Height DP(前缀和)

最新推荐文章于 2022-03-21 11:05:00 发布

orz11111111

最新推荐文章于 2022-03-21 11:05:00 发布

阅读量158

点赞数

分类专栏： CS 泛做 DP

本文链接：https://blog.csdn.net/noone0/article/details/78323731

版权

DP 同时被 2 个专栏收录

53 篇文章 0 订阅

订阅专栏

CS 泛做

34 篇文章 0 订阅

订阅专栏

CS 300
题意:问有多少种形态不同,n个节点,高度为H的二叉树. n,h<=500.
dp[i][j]:i个节点高度为j的二叉树的方法数.

枚举其中一个子树的节点数为k,dp[i][j]+=dp[k][0~j-1]*dp[i-1-k][j-1];
注意两课子树的高度有3种情况 (左=j-1,右<j-1),(左<j-1,右=j-1),(左=右=j-1).
用pre[i][j]记录 dp[i][0]+dp[i][1]+..dp[i][j]. O(N^3) TLE.....要加mod换成减法好像才能过.

法2:设dp[i][j] i个节点高度不超过j的方法数. 最后答案为dp[n][h]-dp[n][h-1].

dp[i][j]= dp[k][j-1]*dp[i-1-k][j-1] [k=0..i-1].省略掉了前缀和化解的步骤.very nice.

TLE代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e2+50,mod=1e9+7;
ll dp[N][N],pre[N][N];
int main()
{
    int n,h;
    scanf("%d%d",&n,&h);
    dp[0][0]=1;
    for(int i=0;i<=h;i++)
        pre[0][i]=1;

    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=h;j++)
        {
            for(int k=0;k<=i-1;k++)
            {
                dp[i][j]=(dp[i][j]+pre[k][j-1]*dp[i-1-k][j-1]%mod)%mod;
                dp[i][j]=(dp[i][j]+dp[k][j-1]*pre[i-1-k][j-1]%mod)%mod;
                dp[i][j]=(dp[i][j]-dp[k][j-1]*dp[i-1-k][j-1]%mod+mod)%mod;
            }
           // cout<<i<<' '<<j<<' '<<dp[i][j]<<endl;
            pre[i][j]=(pre[i][j-1]+dp[i][j])%mod;//
        }
    }
    printf("%lld\n",dp[n][h]);
    return 0;
}

AC代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e2+5,mod=1e9+7;
ll dp[N][N],pre[N][N];
int main()
{
    int n,h;
    scanf("%d%d",&n,&h);
    dp[0][0]=1;
    for(int i=0;i<=h;i++)
        dp[0][i]=1;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=h;j++)
        {
            for(int k=0;k<=i-1;k++)
            {
                dp[i][j]=(dp[i][j]+dp[k][j-1]*dp[i-1-k][j-1]%mod)%mod;
            }
        }
    }
    printf("%lld\n",(dp[n][h]-dp[n][h-1]+mod)%mod);
    return 0;
}