5-28 搜索树判断 (25分)
对于二叉搜索树,我们规定任一结点的左子树仅包含严格小于该结点的键值,而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树,得到的树叫做镜像二叉搜索树。
现在我们给出一个整数键值序列,请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,如果是,则输出对应二叉树的后序遍历序列。
输入格式:
输入的第一行包含一个正整数N(\le≤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 <stdio.h>
#include <stdlib.h>
#include <string.h>
struct node
{
int data;
struct node * left, *right;
};
int pre[12121], que1[12121], que2[12121];
int flag1;
int top1, top2;
struct node * creat(struct node * bt, int k)//建树
{
if(!bt)
{
bt = (struct node *)malloc(sizeof(struct node));
bt->data = k;
bt->left = bt->right = NULL;
return bt;
}
else if(k<bt->data)//左插
bt->left = creat(bt->left, k);
else//右插
bt->right = creat(bt->right, k);
return bt;
};
void swap(struct node * bt)//交换树的左右结点
{
struct node * p = NULL;
if(!bt)//当树的节点为空,返回
return ;
swap(bt->left);//向左递归
swap(bt->right);//向右递归
p = bt->left;//交换
bt->left = bt->right;
bt->right = p;
}
void preprint(struct node * bt)//先序遍历
{
if(bt)
{
que1[++top1] = bt->data;
preprint(bt->left);
preprint(bt->right);
}
}
void postprint(struct node * bt)//后序遍历
{
if(bt)
{
postprint(bt->left);
postprint(bt->right);
que2[++top2] = bt->data;
}
}
int main()
{
int n;
scanf("%d", &n);
for(int i=0;i<n;i++)
scanf("%d", &pre[i]);
struct node *bt = NULL;
for(int i=0;i<n;i++)
{
bt = creat(bt, pre[i]);
}
top1 = -1, top2 = -1;
preprint(bt);
for(int i=0;i<top1;i++)
{
if(pre[i]!=que1[i])//如果不相等,则标记
{
flag1 = 1;
memset(que1, 0, sizeof(que1));
break;
}
}
if(flag1)//如果不是非镜像的
{
swap(bt);//交换
top1 = -1;
preprint(bt);//先序
for(int i=0;i<n;i++)
{
if(pre[i]!=que1[i])//如果镜像也不相等,则标记
{
flag1++;
memset(que1, 0, sizeof(que1));
break;
}
}
}
if(flag1==2)//两者都不可以,NO
{
printf("NO\n");
}
else
{
printf("YES\n");
postprint(bt);//后序
for(int i=0;i<=top2;i++)
{
printf("%d%c", que2[i], i==top2?'\n':' ');
}
}
return 0;
}