L2-004. 这是二叉搜索树吗?
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
一棵二叉搜索树可被递归地定义为具有下列性质的二叉树:对于任一结点,
- 其左子树中所有结点的键值小于该结点的键值;
- 其右子树中所有结点的键值大于等于该结点的键值;
- 其左右子树都是二叉搜索树。
所谓二叉搜索树的“镜像”,即将所有结点的左右子树对换位置后所得到的树。
给定一个整数键值序列,现请你编写程序,判断这是否是对一棵二叉搜索树或其镜像进行前序遍历的结果。
输入格式:
输入的第一行给出正整数N(<=1000)。随后一行给出N个整数键值,其间以空格分隔。
输出格式:
如果输入序列是对一棵二叉搜索树或其镜像进行前序遍历的结果,则首先在一行中输出“YES”,然后在下一行输出该树后序遍历的结果。数字间有1个空格,一行的首尾不得有多余空格。若答案是否,则输出“NO”。
输入样例1:7 8 6 5 7 10 8 11输出样例1:
YES 5 7 6 8 11 10 8输入样例2:
7 8 10 11 8 6 7 5输出样例2:
YES 11 8 10 7 5 6 8输入样例3:
7 8 6 8 5 10 9 11输出样例3:
NO
思路:
题目让你判断给出的序列是否是二叉排序树或其镜像的先序遍历。因为二叉排序的建立过程就是建立在先序遍历的基础上,如果一个值添加到了一个节点的右子树上,下一个值却添加到了左子树上,那么就破坏了先序建立的过程,此时我们对整棵树进行先序遍历得到的序列一定不等于原序列;反之,如果值的添加严格按照先序添加,那么先序遍历得到的序列一定等于原序列。
二叉树的镜像就是将遍历顺序改为先遍历右子树,再遍历左子树。
需要注意镜像树需要对其进行镜像的后序遍历。
代码:
#include<stdio.h>
#include<math.h>
#include<string.h>
#include<stdlib.h>
#include<vector>
#include<ctype.h>
#include<algorithm>
using namespace std;
#define inf 1<<29
int n,arr[1005],cnt,flag,a1[1005],a2[1005];
typedef struct node
{
int data;
node *lch,*rch;
}*Tree;
void add(Tree &T,int d)
{
if(T==NULL)
{
T = new node;
T->data = d;
T->lch = T->rch = NULL;
return;
}
if(d < T->data) add(T->lch,d);
else add(T->rch,d);
}
void preOrder1(Tree &T)
{
if(T)
{
a1[cnt++] = T->data;
preOrder1(T->lch);
preOrder1(T->rch);
}
}
void preOrder2(Tree &T)
{
if(T)
{
a2[cnt++] = T->data;
preOrder2(T->rch);
preOrder2(T->lch);
}
}
void postOrder1(Tree &T)
{
if(T)
{
postOrder1(T->lch);
postOrder1(T->rch);
if(flag) printf(" ");
flag = 1;
printf("%d",T->data);
}
}
void postOrder2(Tree &T)
{
if(T)
{
postOrder2(T->rch);
postOrder2(T->lch);
if(flag) printf(" ");
flag = 1;
printf("%d",T->data);
}
}
int main()
{
scanf("%d",&n);
Tree T = NULL;
for(int i=0; i<n; i++)
{
scanf("%d",&arr[i]);
add(T,arr[i]);
}
cnt=0;
preOrder1(T);
cnt=0;
preOrder2(T);
if(memcmp(arr,a1,n*sizeof(int))==0)
{
printf("YES\n");
flag=0;
postOrder1(T);
exit(0);
}
if(memcmp(arr,a2,n*sizeof(int))==0)
{
printf("YES\n");
flag=0;
postOrder2(T);
exit(0);
}
printf("NO\n");
return 0;
}