7-25 搜索树判断 (25 分)

对于二叉搜索树,我们规定任一结点的左子树仅包含严格小于该结点的键值,而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树,得到的树叫做镜像二叉搜索树。

现在我们给出一个整数键值序列,请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,如果是,则输出对应二叉树的后序遍历序列。

输入格式:
输入的第一行包含一个正整数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

思路:
根据所给的序列创建二叉搜索树和镜像二叉搜索树,
先序遍历所建的树,与所给序列比较,
哪个先序序列与所给序列相同,就后序遍历该序列所对应的树

#include<cstdio>
#include<iostream>
using namespace std;
const int maxn=1005;
struct node{
	int data;
	node* left;
	node* right;
};
typedef node* tree;
int a[maxn],q1[maxn],q2[maxn],t=0;
void swap(tree &T){//交换左右子树
	tree t=NULL;
	if(!T) return ;
	swap(T->left);
	swap(T->right);
	t=T->left;
	T->left=T->right;
	T->right=t;
}
void Insert(tree &T,int x){//二叉搜索树的插入
	if(!T){
		T=new node;
		T->data=x;
		T->left=T->right=NULL;
	}
	else{
		if(x<T->data)
		Insert(T->left,x);
		else if(x>=T->data)
		Insert(T->right,x);
	}
}
void pre(tree T){//先序遍历
	if(!T) return ;
	q1[t++]=T->data;
	pre(T->left);
	pre(T->right);
}
void post(tree T){//后序遍历
	if(!T) return ;
	post(T->left);
	post(T->right);
	q2[t++]=T->data;
}
int main(){
	int n;
	scanf("%d",&n);
	tree T=NULL;
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
		Insert(T,a[i]);
	}
	int flag=0;
	t=0;
	pre(T);//先序遍历二叉搜索树
	for(int i=0;i<n;i++){
		if(a[i]!=q1[i]){
			flag++;
			break;
		}
	}
	if(flag==1){//不是二叉搜索树的先序遍历,先序遍历镜像二叉搜索树
		swap(T);//交换左右子树
		t=0;
		pre(T);//先序遍历镜像二叉搜索树
		for(int i=0;i<n;i++){
			if(a[i]!=q1[i]){
				flag++;
				break;
			}
		}
	}
	if(flag==2)//都不相同
	printf("NO\n");
	else{
		printf("YES\n");
		t=0;
		post(T);
		for(int i=0;i<n-1;i++)
		printf("%d ",q2[i]);
		printf("%d\n",q2[n-1]);
	}
	return 0;
}