PAT甲级 1043 Is It a Binary Search Tree

最新推荐文章于 2024-10-08 20:49:38 发布

CharmsGraker

最新推荐文章于 2024-10-08 20:49:38 发布

阅读量95

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/qq_35332805/article/details/129444103

版权

该文章介绍了一种方法来检查给定的二叉树是否是二叉搜索树或者其镜像。通过升序和降序的中序遍历序列，结合前序遍历，利用二分搜索优化查找过程，处理可能存在的重复值。如果树不合法，则尝试将其视为二叉搜索树的镜像。

摘要由CSDN通过智能技术生成

思路

要检查一颗二叉树是否为二叉搜索树，或者二叉搜索树的镜像。
按照题意，就是把升序、降序的前序序列作为中序分别进行检查。
但是题目涵盖了重复值的节点。所以用unordered_map容器再记录这些值的中序位置不太方便了。

题目说明了重复值必将出现在二叉搜索树的右子树（或二叉搜索树镜像的左子树）
所以在检查二叉搜索树的时候，在中序序列（升序排列）找到首次出现val的索引
对于二叉搜索树镜像，则在中序序列（降序排列）中找到**最后一次出现val**的索引
作为改进，搜索根的中序位置的时候可以使用二分进行加速。

代码

#include<bits/stdc++.h>

using namespace std;

struct TreeNode {
    int val;
    TreeNode*left, *right;
    TreeNode(int v):val(v), left(nullptr), right(nullptr){};
};

vector<int> preorder, inorder;

bool flag = true;
TreeNode * build_tree(int preL, int preR, int inL, int inR, bool mirrorflag = false) {
    if(preL > preR)
        return nullptr;
    int val = preorder[preL];
    auto root = new TreeNode(val);
    int inIdx = -1;
    for(int i = inL; i <= inR; ++i) {
        if(inorder[i] == val) {
            inIdx = i;
            if(! mirrorflag) // 检查二叉搜索树时找第一个，检查镜像时找最后一个
                break;
        }
    }
    // 如果树不合法则返回
    if(inIdx == -1) {
        flag = false;
        return nullptr;
    }
    int n_left = inIdx - inL;
    root->left = build_tree(preL + 1, preL + n_left, inL, inIdx - 1, mirrorflag);
    root->right = build_tree(preL + n_left + 1, preR, inIdx + 1, inR, mirrorflag);
    if(!flag)
        return nullptr;
    return root;
}

TreeNode * pre = nullptr;

void postOrder(TreeNode * root) {
    if(!root)
        return;
    postOrder(root->left);
    postOrder(root->right);
    if(pre)
        printf(" ");
    printf("%d", root->val);
    pre = root;
}

int main() {
    int n;
    cin >> n;
    preorder.resize(n);
    for(int i = 0; i < n; ++i)
        cin >> preorder[i];
    inorder = preorder;
    sort(inorder.begin(), inorder.end());
    auto root = build_tree(0, n - 1, 0, n - 1);
    if(! root) { // 不是二叉搜索树，但有可能是二叉搜索树的镜像，其中序即为降序排列
        reverse(inorder.begin(), inorder.end());
        flag = true; // 重置合法标记
    }
    if(root || (root = build_tree(0, n - 1, 0, n - 1, true))) {
        printf("YES\n");
        postOrder(root);
        printf("\n");
    } else printf("NO\n");
}