平衡二叉树(递推)

平衡二叉树(递推)

打的时候考虑到了把左子树填满,右子树节点数目尽量少,然后在求“节点数目尽量少”的时候思路出错了。

n>d+1时,用递推来解决。考虑高度为n、左右子树节点数目差不超过d的节点最少的树,令它的节点数目为f(n)这棵树的左子树和右子树分别都是平衡的,并且都要节点数目尽量地少,那么我们令左子树节点数目为f(n-1),右子树节点数目为f(n-1-d)就可以满足条件了。

当n<=d+1时,f(n)=n。

按照这个思路写完一直只能过70%样例。然后在别人的代码里面发现 (1ll<<(n-1))这个东西,一查1ll是转成long long型,于是知道自己原来的写法爆int了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll f[65];
int n,d;

int main()
{
    cin>>n>>d;
    ll lchild=0,rchild=0;
    if(n>0)
        lchild=(1ll<<(n-1))-1;
    for(int i=1;i<=n;i++)
    {
        if(i<=d+1)
            f[i]=i;
        else
            f[i]=f[i-1]+f[i-1-d]+1;
    }
    if(n>0)
        rchild=f[n-1-d];
    cout<<lchild-rchild<<endl;
}

 

    原文作者:平衡二叉树
    原文地址: https://blog.csdn.net/xuzonghao/article/details/82935854
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞