5-28 搜索树判断 (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
http://pta.patest.cn/pta/test/15/exam/4/question/843
#include <cstdio>
#include <sstream>
#include <cstring>
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;
#define N 1001
int n ;
int pre[N] ;
typedef struct node{
int data ;
struct node *left , *right ;
node(int _data = -1)
{
data = _data ;
left = NULL ;
right = NULL ;
}
}Bnode;
Bnode* createtree(Bnode* root , int data)
{
if(root == NULL)
{
root = new node(data);
}else{
if(data < root->data)
root->left = createtree(root->left , data);
else
root->right = createtree(root->right , data);
}
return root ;
}
Bnode* createtree2(Bnode* root , int data)
{
if(root == NULL)
{
root = new node(data);
}else{
if(data >= root->data)
root->left = createtree2(root->left , data);
else
root->right = createtree2(root->right , data);
}
return root ;
}
vector<int> v ;
void preOrder(Bnode* root)
{
if(root != NULL)
{
v.push_back(root->data) ;
preOrder(root->left) ;
preOrder(root->right) ;
}
}
void postOrder(Bnode* root)
{
if(root != NULL)
{
postOrder(root->left) ;
postOrder(root->right) ;
v.push_back(root->data) ;
}
}
bool isOk()
{
int i ;
for(i = 0 ; i < n ; i++)
{
if(v[i] != pre[i])
return false ;
}
return true;
}
int main()
{
//freopen("in.txt", "r", stdin);
while(scanf("%d" , &n) != EOF)
{
int i ;
Bnode* root = NULL ;
for(i = 0 ; i < n ; i++)
{
scanf("%d" , &pre[i]) ;
root = createtree(root , pre[i]);
}
v.clear() ;
preOrder(root) ;
if(isOk())
{
printf("YES\n");
v.clear();
postOrder(root) ;
printf("%d" , v[0]) ;
for(i = 1 ; i < n ; i++)
{
printf(" %d" , v[i]);
}
printf("\n") ;
}else{
Bnode* root2 = NULL ;
for(i = 0 ; i < n ; i++)
{
root2 = createtree2(root2 , pre[i]);
}
v.clear() ;
preOrder(root2) ;
if(isOk())
{
printf("YES\n");
v.clear();
postOrder(root2) ;
printf("%d" , v[0]) ;
for(i = 1 ; i < n ; i++)
{
printf(" %d" , v[i]);
}
printf("\n") ;
}else{
printf("NO\n");
}
}
}
return 0 ;
}