1043 判断二叉搜索树
二叉搜索树 (BST) 递归定义为具有以下属性的二叉树:
- 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值
- 若它的右子树不空,则右子树上所有结点的值均大于或等于它的根结点的值
- 它的左、右子树也分别为二叉搜索树
我们将二叉搜索树镜面翻转得到的树称为二叉搜索树的镜像。
现在,给定一个整数序列,请你判断它是否可能是某个二叉搜索树或其镜像进行前序遍历的结果。
输入格式
第一行包含整数 N,表示节点数量。
第二行包含 N 个整数。
输出格式
如果给定整数序列是某个二叉搜索树或其镜像的前序遍历序列,则在第一行输出 YES
,否则输出 NO
。
如果你的答案是 YES
,则还需要在第二行输出这棵树的后序遍历序列。
数据范围
1≤N≤1000
输入样例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
代码实现:
#include<iostream>
#include<vector>
using namespace std;
const int N=1005;
int a[N];
vector<int>v;
void dfs(int l,int r,int flag){
if(l>r)return;
int m=l+1,n=r;
if(flag){
while(m<=r&&a[m]<a[l])m++;
while(n>l&&a[n]>=a[l])n--;
}else{
while(m<=r&&a[m]>=a[l])m++;
while(n>l&&a[n]<a[l])n--;
}
if(n+1!=m)return;
dfs(l+1,n,flag);
dfs(m,r,flag);
v.push_back(a[l]);
}
int main(){
int n;
cin>>n;
for(int i=0;i<n;i++)cin>>a[i];
dfs(0,n-1,1);
if(v.size()!=n){
v.clear();
dfs(0,n-1,0);
}
if(v.size()!=n)cout<<"NO"<<endl;
else{
cout<<"YES"<<endl;
for(int i=0;i<v.size();i++){
if(i==0)cout<<v[i];
else cout<<" "<<v[i];
}
}
return 0;
}