牛客国庆集训派对Day2 F、平衡二叉树 【构造+记忆化搜索】

任意门：https://www.nowcoder.com/acm/contest/202/F

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 1048576K，其他语言2097152K
64bit IO Format: %lld

题目描述 

平衡二叉树，顾名思义就是一棵“平衡”的二叉树。在这道题中，“平衡”的定义为，对于树中任意一个节点，都满足左右子树的高度差不超过 d. 空树的高度定义为0，单个节点的高度为1，其他情况下树的高度定义为根节点左右子树高度最大值 + 1. 一棵在高度上平衡的树，节点数可能不平衡，因此再定义一棵树的不平衡度为这棵树中所有节点的左右子树的节点数之差的最大值。
给定平衡的定义参数d, 你需要求出所有高度为 n 的平衡树中不平衡度的最大值。

输入描述:

两个整数，n, d.

输出描述:

一个整数：所有高度为 n 的平衡树中不平衡度的最大值。

示例1

输入

复制

4 1

输出

复制

5

说明

下面这棵树在 d=1 的定义下高度是平衡的，其不平衡度为 5。

 

备注:

0 ≤ n, d ≤ 60

题意概括：

不要被题目蒙骗了，这是一道找规律和构造的题目。

tip：注意是所有左右子树的高度差为 d，不平衡度也是所有左右子树比较后的最大的那个。

解题思路：

思路很清晰嘛，第一个左结点为根的子树为满二叉树，第一个右结点的子树以尽量少的结点数构成一颗满足条件的二叉树，答案为这两颗子树的结点之差。

第一个左子树为满二叉树，左子树结点数 2^(N-1)-1; （快速幂）

那右边的那一半呢就进行构造建树吧，具体实现为 dfs ，每次构造它的左子树和右子树，直到达到需要的高度。但是一开始纯dfs是肯定超时，这时候需要记忆化搜索优化一下。

AC code：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#define LL long long
using namespace std;
LL dp[100][100];
int N,d;

LL dfs(int h, int step)
{
    if(dp[h][step] != -1) return dp[h][step];
    LL sum = 1LL;
    if(step == h) {dp[h][step] = sum; return sum;}
    sum += dfs(h, step+1);                  //左子树
    if(step >= h-d) {dp[h][step] = sum; return sum;}
    sum += dfs(h-d, step+1);  //右子树
    dp[h][step] = sum;
    return sum;
}

LL qpow(int x, int n)
{
    LL ans = 1, base = x;
    while(n != 0){
        if(n&1!=0) ans*=base;
        base*=base;
        n>>=1;
    }
    return ans;
}

int main()
{
    //d = 1;
    //printf("%d
", dfs(3, 1));
    LL L, R;
    //while(~scanf("%lld%lld", &N, &d)){
    scanf("%d%d", &N, &d);
    memset(dp, -1, sizeof(dp));
    if(N == 0 || d == 0) L = 0, R = 0;
    else{
        L = qpow(2, N-1)-1;
        if(d >= N-1 ) R = 0;
        else  R = dfs(N-d, 2);
    }
    //printf("L:%lld R:%lld
", L, R);
    printf("%lld
", L-R);
    //}
    return 0;
}