1043 Is It a Binary Search Tree

最新推荐文章于 2021-12-14 13:26:38 发布

原创最新推荐文章于 2021-12-14 13:26:38 发布 · 120 阅读

0 ·

CC 4.0 BY-SA版权

SHUATI 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种算法，用于判断给定序列是否为二叉搜索树的前序遍历序列。通过构建BST并进行前序遍历，比较生成的序列与给定序列是否一致。同时，考虑到镜像BST的情况，输出相应的后序遍历序列。

题目大意：给定一个查找二叉树的序列，判定这个序列是不是前序遍历的对应的BST的前序遍历序列。

思路：考察基本的树的建立和遍历等知识。

注意点：如果是镜像BST，要输出镜像的后序遍历。（题目好像没有说清楚）

CODE ⇩⇩⇩

#include <bits/stdc++.h>
using namespace std;
vector<int> preList,mirrorPreList,postList,mirrorPostList;
int n;
struct Node{
    int data;
    Node *left,*right;
    explicit Node(int data){this->data = data;this->right = nullptr;this->left = nullptr;}
};
void insert(Node* &root,int i){
    if(root == nullptr) {
        Node *node = new Node(i);
        root = node;
    }
    else if(i < root->data) insert(root->left,i);
    else insert(root->right,i);
}
void preOrder(Node *node){
    if(node == nullptr) return;
    preList.push_back(node->data);
    preOrder(node->left);
    preOrder(node->right);
}
void mirrorPreOrder(Node *node){
    if(node == nullptr) return;
    mirrorPreList.push_back(node->data);
    mirrorPreOrder(node->right);
    mirrorPreOrder(node->left);
}
void postOrder(Node *root){
    if(root == nullptr) return;
    postOrder(root->left);
    postOrder(root->right);
    postList.push_back(root->data);
}
void mirrorPostOrder(Node *root){
    if(root == nullptr) return;
    mirrorPostOrder(root->right);
    mirrorPostOrder(root->left);
    mirrorPostList.push_back(root->data);
}
void output(vector<int> vc){
    cout<<vc.front();
    for(int i =1;i < n;i++) cout<<" "<<vc[i];
}
int main() {
    cin>>n;
    vector<int> lis(n);
    for(int i = 0;i < n;i++) cin>>lis[i];
    Node *root = nullptr;
    for(int i = 0;i < n;i++) insert(root,lis[i]);
    preOrder(root),mirrorPreOrder(root);
    if(lis == preList){
        cout<<"YES"<<endl;
        postOrder(root);
        output(postList);
    }else if(lis == mirrorPreList){
        cout<<"YES"<<endl;
        mirrorPostOrder(root);
        output(mirrorPostList);
    }else{
        cout<<"NO"<<endl;
    }
    return 0;
}