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

牛客国庆集训派对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;
}