牛客国庆集训派对Day2 平衡二叉树 DP找规律

最新推荐文章于 2021-02-13 22:08:49 发布

原创最新推荐文章于 2021-02-13 22:08:49 发布 · 343 阅读

0 ·

CC 4.0 BY-SA版权

比赛专栏收录该内容

8 篇文章

订阅专栏

探讨了求解最小深度差为d的平衡二叉树的最少节点数问题，通过动态规划算法实现，适用于算法竞赛及数据结构学习。

牛客国庆集训派对Day2 平衡二叉树

求最小深度差为d的平衡树的最少节点数

#include<bits/stdc++.h>
using namespace std;
const int MAX=61;
int n,d;
long long dp[MAX];
int main()
{
    scanf("%d%d",&n,&d);
    long long l=(1ll<<(n-1))-1,r;
    dp[0]=0;
    for(int i=1;i<=n-d-1;++i)
    {
        if(i<=d+1) dp[i]=i;
        else dp[i]=dp[i-1]+dp[i-d-1]+1;
    }
    r=dp[max(0,n-d-1)];
    printf("%lld\n",l-r);
    return 0;
}