L2-004 这是二叉搜索树吗？——重建二叉树解答

题目

L2-004 这是二叉搜索树吗？ (25 分)

一棵二叉搜索树可被递归地定义为具有下列性质的二叉树：对于任一结点，

• 其左子树中所有结点的键值小于该结点的键值；
• 其右子树中所有结点的键值大于等于该结点的键值；
• 其左右子树都是二叉搜索树。

所谓二叉搜索树的“镜像”，即将所有结点的左右子树对换位置后所得到的树。

给定一个整数键值序列，现请你编写程序，判断这是否是对一棵二叉搜索树或其镜像进行前序遍历的结果。

输入格式：

输入的第一行给出正整数 N（≤1000）。随后一行给出 N 个整数键值，其间以空格分隔。

输出格式：

如果输入序列是对一棵二叉搜索树或其镜像进行前序遍历的结果，则首先在一行中输出 YES ，然后在下一行输出该树后序遍历的结果。数字间有 1 个空格，一行的首尾不得有多余空格。若答案是否，则输出 NO。

输入样例 1：

7
8 6 5 7 10 8 11

输出样例 1：

YES
5 7 6 8 11 10 8

输入样例 2：

7
8 10 11 8 6 7 5

输出样例 2：

YES
11 8 10 7 5 6 8

输入样例 3：

7
8 6 8 5 10 9 11

输出样例 3：

NO

---

样例1 所得二叉树如下

代码

#include <iostream>
#include <vector>
#include <stdio.h>

using namespace std;

struct TreeNode
{
    TreeNode* left;
    TreeNode* right;
    int val;
};

bool check(vector<bool> ls){
    int count = ls.size();
    int cnt = 0;
    for (int i = 1; i < count; i++)
    {
        if (ls[i-1] != ls[i]) cnt++;
        if (cnt > 1) return false;
    }
    return true;
}

bool build(vector<int> f_ord, TreeNode* root, bool left_smaller){
    int size = f_ord.size();
    root->val = f_ord[0];
    if (size == 1)
    {
        root->left = nullptr;
        root->right = nullptr;
        return true;
    }    
    TreeNode* l = new TreeNode();
    TreeNode* r = new TreeNode();
    vector<int> l_f_ord,r_f_ord;
    vector<bool> to_left(size, false);
    for (int i = 1; i < f_ord.size(); i++)
    {
        if ((f_ord[i] < root->val) == left_smaller)
        {
            r_f_ord.push_back(f_ord[i]);  // 左子树    
            to_left[i] = true;   
        }
        else
        {
            l_f_ord.push_back(f_ord[i]); // 右子树        
            to_left[i] = false;
        }
    }
    //检查是否合法，to_left应该形如1 1 1 0 0 0 或者 0 0 0 1 1 1
    if (!check(to_left)) return false; 
    if (!l_f_ord.empty())
    {
        bool b = build(l_f_ord, l, left_smaller);
        if (!b) return false;
    }
    else l = nullptr;
    if (!r_f_ord.empty())
    {
        bool b = build(r_f_ord, r, left_smaller);
        if (!b) return false;
    }
    else r = nullptr;
    root->left = l;
    root->right = r;
    return true;
}

void visit(TreeNode* root){
    // 后序遍历
    if (root->left) visit(root->left);
    if (root->right) visit(root->right);
    cout << root->val << ' ';
}

int main()  
{
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout); 
    
    int n;
    cin >> n;

    vector<int> first_ord(n, 0);
    for (int i = 0; i < n; i++)
    {
        cin >> first_ord[i];
    }

    TreeNode* root = new TreeNode();
    bool lsmall = n > 1? first_ord[0] < first_ord[1]: true;
    if (build(first_ord, root, lsmall)) 
    {
        cout << "YES" << endl;
        if (root->left) visit(root->left);
        if (root->right) visit(root->right);
        cout << root->val;
    }
    else cout << "NO";

    return 0;
}