1064 Complete Binary Search Tree (30 分)*

原题目： 1064 Complete Binary Search Tree (30 分).

题意

二叉查找树的左子树结点值比根结点小，右子树结点值大于或等于根结点。给出结点数N以及N个不重复的非负整数（不大于2000）。建立二叉查找树，同时也是完全二叉树，
👉输出它的层序序列。

分析

已知中序序列，可求root下标，进而求出层序。

1. 因为二叉搜索树BST的中序满足——是一组递增的有序序列，所以只需排序所给序列即可得到中序。
2. 根据完全二叉树的结点数，可以求出它的根结点在中序中对应的下标——用DFS递归，需要中序序列区间[inl, inr] 以及 完全二叉树下标 index，则计算 完全二叉树根结点在中序序列中下标 如下👇(均从0开始编号）
   ① 结点数 n = inl - inr + 1
   ② 树高 l = log₂(n+1)
   ③ 最后一层叶结点数 leaf = n - (2^l - 1)
   ④ 计算根结点在中序中的下标 root = inl + (2^(l-1)- 1) + min(2^(l-1), leaf)

CODE

#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
using namespace std;
vector<int> in, level;
void levelOrder(int inl, int inr, int index);
int main()
{
    int n;
    cin >> n;
    in.resize(n), level.resize(n);
    for ( int i=0; i<n; i++ )
        cin >> in[i];
    sort(in.begin(), in.end());  //BST的中序序列即递增序列
    //生成层序
    levelOrder(0, n-1, 0);
    cout << level[0];
    for ( int i=1; i<n; i++ )
        cout << " " << level[i];
    
    return 0;
}
void levelOrder(int inl, int inr, int index){
    if ( inl>inr ) return;
    //求根结点在中序中对应的下标
    int n = inr - inl + 1;  //以[inl,inr]为区间的完全二叉树的结点数
    int l = log(n+1) / log(2);  //即 树高-1 =log(2)(n+1) （除去最后一层的层数）
    int leaf = n - (pow(2,l) - 1);  //最后一层的叶结点数（总结点数-除最后一层外的所有结点数）
    int root = inl + (pow(2,l-1) - 1) + min((int)pow(2,l-1), leaf);
    //按完全二叉树的下标顺序，进行先序遍历即得层序序列
    level[index] = in[root];
    levelOrder(inl, root-1, 2*index+1);  //左子树
    levelOrder(root+1, inr, 2*index+2);  //右子树
}