递推
要求所有高度为n的平衡树中不平衡度的最大值,没必要考虑所有的树,只要想办法构造出一棵不平衡度最大的树即可。
要想不平衡度最大,只要根节点的左右子树节点数差值最大即可。若不选取根节点,则高度上限会减少,得到的树不平衡度不是最大。而要使左右子树节点数差值最大,只要使左子树节点数最大、右子树节点数尽可能小即可。
左子树节点数最大即排满,为 -1。
考虑右子树,即考虑在n和d的限制下节点数最少的平衡树,这里采用递推的方法。
记f[n]为高度为n的节点数最少的平衡树,其左右子树高度差不超过d。要构造高度为n的树,可以进行如下操作:
新建一个节点,其左子树为高度为n-1的平衡树,其右子树为高度为n-1-d的平衡树,该子树即为满足条件的高度为n的树。
转化为递推公式即为:f[n]=f[n-1]+1+f[n-1-d]。(当n-1-d<0时,右子树为空树)
新建节点连接高度为n-1的树保证了此树的高度为n;新建节点连接高度为n-1-d的树保证了此树为平衡树;而新建节点连接的两个数本身节点数最小保证了此树的节点数最小。
答案即为 -1-f[n-1-d](n-1-d<0时,右子树为空树)。
代码:
#include <iostream> #include <queue> #include <set> #include <map> #include <vector> #include <stack> #include <cmath> #include <algorithm> #include <cstdio> #include <cctype> #include <functional> #include <string> #include <cstring> #include <sstream> #include <deque> #define fir first #define sec second using namespace std; typedef long long ll; typedef pair<int,int> P; typedef pair<P,int> Q; const int inf1=2e9+9; const ll inf2=8e18+9; const ll mol=1e9+7; const int maxn=1e5+9; const ll maxx=1e12+9; int n,d; ll f[66]={0,1}; ll q_pow(ll x,ll t) { if(t<0) return 1; ll res=1; while(t) { if(t&1) res*=x; x*=x; t>>=1; } return res; } int main() { scanf("%d%d",&n,&d); for(int i=2;i<=n;i++) f[i]=f[i-1]+1+f[max(i-1-d,0)]; printf("%lld",q_pow(2,n-1)-1-f[max(n-1-d,0)]); }