E. 搜索树判断

目录

题目描述

思路分析

AC代码

---

题目描述

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

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

输入

输入的第一行包含一个正整数N（≤1000），第二行包含N个整数，为给出的整数键值序列，数字间以空格分隔。

输出

输出的第一行首先给出判断结果，如果输入的序列是某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列，则输出YES，否侧输出NO。如果判断结果是YES，下一行输出对应二叉树的后序遍历序列。数字间以空格分隔，但行尾不能有多余的空格。

输入样例

7
8 6 5 7 10 8 11

输出样例

YES
5 7 6 8 11 10 8

思路分析

其实一开始没什么思路，是看了下面这篇文章才写出来的。 就是说，如果它是搜索二叉树的话，第一个是根结点，从第二个到比根结点大的之前，都是左子树。比根节点大的到最后都是右子树。然后再递归判断左子树和右子树。镜像的同理。

7-3 搜索树判断（7-4 这是二叉搜索树吗？）_Lurume的博客-CSDN博客

AC代码

#include<iostream>
using namespace std;

int a[1000],ans[1000];
int n,k=0;

int if_tree(int start,int end){
	int tag;
	if(start>=end) return 1;
	
	int center=start+1;
	while(a[center]<a[start]){
		if(center>=n) break;
		center++;
	}
	
	for(int i=center;i<end;i++){
		if(a[i]<a[start])
		return 0;
	}
	
	ans[k++]=a[start];
	
	tag=if_tree(center,end);   //右子树 
	if(tag==0) return 0;
	
	tag=if_tree(start+1,center);    //左子树 
	if(tag==0) return 0;
	return 1;
}

int if_mirror_tree(int start,int end){
	
	    int tag;
		if(start >= end) return 1;
		int center = start+1;
		while(a[center] >= a[start])
		{
	        if(center >= n) break;
			center ++;
		}
		for(int i = center;i < end;i ++)//检查是否合规
		{
			if(a[i] >= a[start])
				return 0;
		}
		ans[k++] = a[start];//后序遍历的倒序存放到ans数组中
		tag = if_mirror_tree(center,end);//优先遍历右子树
	    if(tag == 0)return 0;
		tag = if_mirror_tree(start+1,center);//左子树
	    if(tag == 0)return 0;
		return 1;
} 

int main() {

	int tag; 
	cin >> n;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	if(a[0]>a[1]){    //也许是二叉搜索树 
//	cout<<"二叉搜索树"<<endl; 
		tag=if_tree(0,n);
	}
	else {                 //也许是镜像 
//		cout<<"镜面"<<endl;
		tag=if_mirror_tree(0,n);
		
	} 
	if(tag==0) cout<<"NO"<<endl;
	else{
		cout<<"YES"<<endl;
		cout<<ans[k-1];
		for(int i=k-2;i>=0;i--){
			cout<<" "<<ans[i];
		}
	} 
	
	
	return 0;
}

有错误欢迎指正！