PAT 1043 Is It a Binary Search Tree

题目描述

分析：

给出N个正整数插入一棵二叉排序树，问这串序列是否是该二叉排序树的先序序列或者是该二叉排序树的镜像树的先序序列，输出后序序列。构建二叉排序树，镜像树的先序遍历只需要交换左右子树的访问顺序即可。

#include<cstdio>
#include<vector>
using namespace std;
struct node{
	int data;
	node* left,*right;
};

void insert(node* &root,int data){
	if(root==NULL){
		root=new node;
		root->data=data;
		root->left=root->right=NULL;
		return;
	}
	if(data<root->data) insert(root->left,data);
	else insert(root->right,data);
} 
void preOrder(node* root,vector<int>& v){
	if(root==NULL) return ;
	v.push_back(root->data);
	preOrder(root->left,v);
	preOrder(root->right,v);
}
void preOrderMirror(node* root,vector<int>& v){
	if(root==NULL) return ;
	v.push_back(root->data);
	preOrderMirror(root->right,v);
	preOrderMirror(root->left,v);
}
void postOrder(node* root,vector<int>& v){
	if(root==NULL) return;
	postOrder(root->left,v);
	postOrder(root->right,v);
	v.push_back(root->data);
}
void postOrderMirror(node* root,vector<int>& v){
	if(root==NULL) return;
	postOrderMirror(root->right,v);
	postOrderMirror(root->left,v);
	v.push_back(root->data);
}
vector<int> origin,pre,preM,post,postM;
int main(){
	int n,data;
	node* root=NULL;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&data);
		origin.push_back(data);
		insert(root,data);
	}
	preOrder(root,pre);
	preOrderMirror(root,preM);
	postOrder(root,post);
	postOrderMirror(root,postM);
	if(origin==pre){
		printf("YES\n");
		for(int i=0;i<post.size();i++){
			printf("%d",post[i]);
			if(i<post.size()-1) printf(" ");
		}
	}else if(origin==preM){
		printf("YES\n");
		for(int i=0;i<postM.size();i++){
			printf("%d",postM[i]);
			if(i<postM.size()-1) printf(" ");
		}
	}
	else{
		printf("NO\n");
	}
}