⭐⭐⭐⭐⭐【重构BST】PAT A1043 Is It a Binary Search Tree (25分)

疑问

暂无

代码

//二分查找树
//需要注意的是，二分查找树的先序遍历序列，应该能重新构建这棵二分查找树 
#include<cstdio>

struct node{
	int data;
	node* lchild;
	node* rchild;
};

const int maxn = 1010;

int input[maxn];

//前序数组 
int pre[maxn];
int m=0;

//后序数组 
int post[maxn];
int p=0;

node* newNode(int v){
	node* root = new node;
	root->data = v;
	root->lchild = NULL;
	root->rchild = NULL;
	return root;
}

void insert(node* &root,int x){
	if(root == NULL){
		root = newNode(x);
	}else if(x < root->data){
		insert(root->lchild,x);
	}else if(x >= root->data){
		insert(root->rchild,x);
	}
}

node* create(int data[],int n){
	node* root = NULL;
	for(int i=0;i<n;i++){
		insert(root,data[i]);
	}
	return root;	
}

void preOrder(node* root){
	if(root == NULL){
		return;
	}else{
		pre[m++] = root->data;
		preOrder(root->lchild);
		preOrder(root->rchild);
	}
} 

void postOrder(node* root){
	if(root == NULL){
		return;
	}else{
		postOrder(root->lchild);
		postOrder(root->rchild);
		post[p++] = root->data;
	}
} 

int main(){
	int n;
	scanf("%d",&n);
	
	//用于记录是否满足
	bool flag = true; 
	
	for(int i=0;i<n;i++){
		scanf("%d",&input[i]);
	}
	
	//根据输入的序列建立一棵二叉查找树 
	node* root = NULL;
	
	root = create(input,n);
	
	//通过先序遍历或者后序遍历得到的二叉查找树来检查是否符合条件 
	preOrder(root);
	
	for(int i=0;i<n;i++){
		if(pre[i] != input[i]){
			flag = false;
			break;
		}
	}
	
	if(flag == true){
		printf("YES\n");
		
		postOrder(root);
		
		for(int i=0;i<n;i++){
			printf("%d%c",post[i],i!=n-1?' ':NULL);
		}
		
		return 0;
	}else{
		flag = true;
		
		postOrder(root);
		
		for(int i=0;i<n;i++){
			if(post[n-i-1] != input[i]){
				flag = false;
				break;
			}
		}
		
		if(flag == true){
			printf("YES\n");
			//直接使用之前的inOrder进行输出
			for(int i=0;i<n;i++){
				printf("%d%c",pre[n-i-1],i!=n-1?' ':NULL);
			}
			return 0; 
		}
	}
	
	printf("NO");
	return 0;	
} 

反思

1. 这道题我第一次写的时候，连怎么动手都不会，完全不知道对于一棵BST而言，它的前序遍历序列是可以重构它的这一特点，后来看了《算法笔记》上的解答才恍然大悟，我还菜。。。
2. 至于mirror image of BST，只需要注意它与BST之间的对应就好啦，自己举几个小例子就明白了。

二刷代码

二刷知道了前序可以构建BST，但是还是没有意识到BST和它的对称树之间的关系：

mirror image of BST的preOrder是BST的postOrder的逆；
mirror image of BST的postOrder是BST的preOrder的逆；

这个结论其实在LeetCode上的某道题目上就有出现过！只是当时自己去暴力解题了。

//使用对称树的一些性质进行解题 
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1010; 

struct node{
	int data;
	node* lchild;
	node* rchild;
	node(){}
	node(int x){
		data=x;
		lchild=NULL;
		rchild=NULL;
	}
}; 

int n;//表示结点的数量 

int pre[maxn];
int preCnt=0;

int in[maxn];
int inCnt=0;

int post[maxn];
int postCnt=0;

int input[maxn];

void insert(node* &root,int x){
	if(root==NULL){
		root = new node(x);//这里用的是应用 
	}else{
		if(root->data > x){
			insert(root->lchild,x);
		}else{
			insert(root->rchild,x);
		}
	}
}

node* create(){
	node* root=NULL;
	for(int i=0;i<n;i++){
		insert(root,input[i]);
	}
	return root;
}

void preOrder(node* root){
	if(root==NULL){
		return;
	}else{
		pre[preCnt++] = root->data;
		preOrder(root->lchild);
		preOrder(root->rchild);
	}
}

void postOrder(node* root){
	if(root==NULL){
		return;
	}else{
		postOrder(root->lchild);
		postOrder(root->rchild);
		post[postCnt++] = root->data;
	}
}

int main(){
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&input[i]); 
	}
	
	//接下来利用输入进行构树
	node* root = create(); 
	
	//进行前序遍历和后序遍历
	preOrder(root);
	postOrder(root);
	
	//printf("%d %d\n",preCnt,postCnt);
	//开始进行匹配，首先检查是否是原有二叉树
	bool flag = true;
	for(int i=0;i<n;i++){
		if(input[i] != pre[i]){
			flag = false;
			break;
		}
	}
	
	if(flag == true){
		printf("YES\n");
		//那么直接输出后序结果
		for(int i=0;i<n;i++){
			printf("%d",post[i]);
			if(i!=n-1){
				printf(" ");
			}
		}
		return 0;  
	}else{
		//判断是否是镜像树
		flag = true;
		for(int i=0;i<n;i++){
			if(input[i]!=post[n-1-i]){
				flag=false;
			}
		}
		
		if(flag == true){
			printf("YES\n");
			//那么输出后序的结果
			for(int i=0;i<n;i++){
				printf("%d",pre[n-1-i]);
				if(i!=n-1){
					printf(" ");
				}
			}
			return 0; 
		} 
	}
	
	printf("NO\n");
	
	return 0;
}