平衡二叉树
平衡二叉树
https://ac.nowcoder.com/acm/problem/19775
思路:根据题意我们让左子树为高度n的满二叉树,右子树上的点尽可能少,那么我们首先吧他的高度拉满使其为n-m-1,此时我们在右子树的左子树上高度拉满点最少,然后如果还有点,那么为了维持平衡且让右子树点最少,我们吧点加到i-m+1的树上,可得方程dp[i]=dp[i-1]+dp[i-m-1]。dp[i]表示高度为i的右子树的最少点。
#include <cstdio> #include <cstring> #include <algorithm> #include <set> #include<iostream> #include<vector> //#include<bits/stdc++.h> using namespace std; typedef long long ll; #define SIS std::ios::sync_with_stdio(false) #define space putchar(' ') #define enter putchar('\n') #define lson root<<1 #define rson root<<1|1 typedef pair<int,int> PII; const int mod=1e4+7; const int N=2e5+10; const int inf=0x7f7f7f7f; ll gcd(ll a,ll b) { return b==0?a:gcd(b,a%b); } ll lcm(ll a,ll b) { return a*(b/gcd(a,b)); } template <class T> void read(T &x) { char c; bool op = 0; while(c = getchar(), c < '0' || c > '9') if(c == '-') op = 1; x = c - '0'; while(c = getchar(), c >= '0' && c <= '9') x = x * 10 + c - '0'; if(op) x = -x; } template <class T> void write(T x) { if(x < 0) x = -x, putchar('-'); if(x >= 10) write(x / 10); putchar('0' + x % 10); } ll dp[N]; int main() { int n,m; scanf("%d%d",&n,&m); dp[1]=1; for(int i=2;i<=n-m+1;i++) { dp[i]=dp[i-1]+1; if(i>m) { dp[i]+=dp[i-m-1]; } } ll ans=(1ll<<(n-1))-1-dp[n-m-1]; cout<<ans<<endl; return 0; }