1043 Is It a Binary Search Tree (25 分)
题目传送门:1043 Is It a Binary Search Tree (25 分)
一、题目大意
判断给定的序列是否可以构成一个二叉查找树或镜像二叉查找树。
二、解题思路
递归判断指定的区间序列是否是二叉查找树的先序。
区间[left, right]中,left是先序二叉树的根,设p为左子树的右端点,q为右子树的右端点,如果可以构成二叉查找树或镜像,那么一定有[left+1, p]是小于根,[p+1, right]大于等于根。如果是反过来的,则是镜像。如果p!=right,则表示不能构成二叉查找树或镜像,则终止递归
边递归边将访问到的根节点存到集合里去。存入集合时的顺序就是遍历二叉树的顺序,题目要求输出后序遍历,那就在递归左右子树后将根存入后序集合中即可。而判断是否是一个二叉查找树,则只需判断后序集合的数量是否等于给定的先序集合数量。如果不等,则说明某个子树p!=right,提前终止了递归,无法构成二叉查找树或镜像。
三、AC代码
#include<bits/stdc++.h>
using namespace std;
template <typename T = int>
T read(){
T x;
cin >> x;
return x;
}
int type;// 0表示BST, 1表示镜像BST
vector<int>preOrder, postOrder;
void BST(int left, int right){
if(left > right)return;
int p = left;// p为左子树的终点
for(int i = left + 1; i <= right; i++){
if(type? preOrder[i] < preOrder[left]: preOrder[i] >= preOrder[left])break;
p = i;
}
int q = p;// q为右子树的终点
for(int i = p + 1; i <= right; i++){
if(type? preOrder[i] >= preOrder[left]: preOrder[i] < preOrder[left])break;
q = i;
}
if(q != right)return;// 无法构成二叉查找树或镜像
BST(left+1, p);
BST(p+1, right);
postOrder.push_back(preOrder[left]);
}
int main(){
int n = read();
for(int i = 0; i < n; i++){
preOrder.push_back(read());
}
type = 0;
BST(0, n-1);
if(postOrder.size() != n){
type = 1;
BST(0, n-1);
}
if(postOrder.size() != n){
cout << "NO" << endl;
return 0;
}
cout << "YES" << endl;
for(int i = 0; i < postOrder.size(); i++){
if(i)cout << " " << postOrder[i];
else cout << postOrder[i];
}
cout << endl;
}