对于二叉搜索树,我们规定任一结点的左子树仅包含严格小于该结点的键值,而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树,得到的树叫做镜像二叉搜索树。
现在我们给出一个整数键值序列,请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,如果是,则输出对应二叉树的后序遍历序列。
输入格式:
输入的第一行包含一个正整数N(≤1000),第二行包含N个整数,为给出的整数键值序列,数字间以空格分隔。
输出格式:
输出的第一行首先给出判断结果,如果输入的序列是某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,则输出
YES
,否侧输出NO
。如果判断结果是YES
,下一行输出对应二叉树的后序遍历序列。数字间以空格分隔,但行尾不能有多余的空格。
思路 :
一、先按照搜索树的特点,构造出一段前序遍历序列
例如序列213,插入2时直接插入即可
插入1时与2比较,发现严格小于2,往左边插入
插入3时,发现严格大于2,往右边插入
镜像的也是一样的处理套路
二、对构造的树求前序遍历,与前面的前序序列比对即可
代码如下
#include<stdio.h>
#include<iostream>
#include<stdlib.h>
#include<vector>
using namespace std;
struct Tnode{
int val;
Tnode*left;
Tnode*right;
};
int num[1010];
vector<int>B;
vector<int>res;
Tnode*Insert(Tnode*p,int x,int flag){
if(p==NULL){
Tnode* T=(Tnode*)malloc(sizeof(Tnode));
T->val=x;
T->left=T->right=NULL;
return T;
}
else {
if(flag==0){
if(x>=p->val) p->right=Insert(p->right,x,flag);
else p->left=Insert(p->left,x,flag);
}
else {
if(x<p->val) p->right=Insert(p->right,x,flag);
else p->left=Insert(p->left,x,flag);
}
}
return p;
}
void preTraver(Tnode*p){
if(p==NULL) return;
B.push_back(p->val);
preTraver(p->left);
preTraver(p->right);
}
void neTraver(Tnode*p){
if(p==NULL) return;
neTraver(p->left);
neTraver(p->right);
res.push_back(p->val);
}
int main(){
int n;
cin>>n;
int flag=0;
int matched=1;
Tnode*T=NULL;
for(int i=1;i<=n;i++){
cin>>num[i];
}
if(n>1&&num[2]>=num[1]) flag=1;
for(int i=1;i<=n;i++){
T=Insert(T,num[i],flag);
}
preTraver(T);
for(int i=1;i<=n;i++){
if(num[i]!=B[i-1]){
matched=0;
break;
}
}
if(matched==1){
neTraver(T);
printf("YES\n");
for(int i=0;i<n;i++){
if(i!=0) printf(" ");
printf("%d",res[i]);
}
}
else printf("NO");
return 0;
}