题目描述
对于二叉搜索树,我们规定任一结点的左子树仅包含严格小于该结点的键值,而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树,得到的树叫做镜像二叉搜索树。
现在我们给出一个整数键值序列,请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,如果是,则输出对应二叉树的后序遍历序列。
输入格式:
输入的第一行包含一个正整数N(≤1000),第二行包含N个整数,为给出的整数键值序列,数字间以空格分隔。
输出格式:
输出的第一行首先给出判断结果,如果输入的序列是某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,则输出YES,否侧输出NO。如果判断结果是YES,下一行输出对应二叉树的后序遍历序列。数字间以空格分隔,但行尾不能有多余的空格。
输入样例1:
7
8 6 5 7 10 8 11
输出样例1:
YES
5 7 6 8 11 10 8
输入样例2:
7
8 6 8 5 10 9 11
输出样例2:
NO
题解
画图观察二叉搜索树性质:
镜像:
我们发现从root开始,直到第一个大于root的数的序列是左子树,,后面的是右子树,镜像反之
我们用两个指针,tl指向l,如果当前数小于root,tl++;
tr指向右边,然后根据二叉搜索树的性质,如果当前数大于等于root,tr–;
如果tr-tl==1,那么成立,算法重复进行
代码
#include<bits/stdc++.h>
using namespace std;
int n;
int pre[2000];
bool A;
int cnt=0;
queue<int> q;
void solve(int l,int r){
if(r<l) return ;
int tl=l+1,tr=r;
int root=pre[l];
if(!A){
while(tl<=r&&pre[tl]<root) tl++;
while(tr>l&&pre[tr]>=root) tr--;
}else{
while(tl<=r&&pre[tl]>=root) tl++;
while(tr>l&&pre[tr]<root) tr--;
}
//cout<<tr<<" tl="<<tl<<" cnt="<<cnt<<endl;
if(tl-tr!=1) return;
solve(l+1,tl-1);
solve(tl,r);
cnt++;
//记录后序遍历顺序
q.push(root);
}
int main(){
cin>>n;
for(int i=1;i<=n;i++) cin>>pre[i];
solve(1,n);
//cout<<cnt;
if(cnt!=n){
cnt=0;
A=1;//镜像情况
solve(1,n);
}
if(cnt!=n){
cout<<"NO";
}else{
cout<<"YES"<<endl;
cout<<q.front();
q.pop();
while(!q.empty()){
int k=q.front();
q.pop();
cout<<" "<<k;
}
}
return 0;
}