题意:给一个二叉树序列的preoder,判断这个二叉树是不是BST(二叉搜索树),或者是不是它的镜像树;如果是,就输出它的后序遍历postorder,否则输出NO
难点:刚开始对序列进行Insert建树,在先序遍历,再匹配,得到了NO,没过样例以为一个BST建树有多种建发,以为自己思路不对;后来问了同学,才晓得题目还有一个镜像树,条件是只要 “or” 一个情况为BST即可的。
PS:我这个思路是纯按着题意去模拟,柳神写的代码就非常短精悍高效,太厉害了!
Code:
#include<bits/stdc++.h>
using namespace std;
#define inf 1009
#define INF 0x3f3f3f3f
int n;
int e[inf],id;//输入preorder以及遍历下标
bool flag;//结果标记
struct Node
{
int value;
Node *left,*right;
Node(int i)
{
value=i;
left=right=NULL;
}
}*root;
void insert(Node *h,int v)
{
if(v<h->value)
{
if(h->left==NULL)
{
Node *node=new Node(v);
h->left=node;
}
else insert(h->left,v);
}
else
{
if(h->right==NULL)
{
Node *node=new Node(v);
h->right=node;
}
else insert(h->right,v);
}
}
void preorder(Node *h)
{
if(!flag||h==NULL)return;
if(e[id++]!=h->value)
{//printf("debugggggg %d %d\n",h->value,e[id-1]);
flag=false;
return;
}
preorder(h->left);
preorder(h->right);
}
void swap_Tree(Node *h)
{
if(h==NULL)return;
swap_Tree(h->left);
swap_Tree(h->right);
Node *p=h->left;
h->left=h->right;
h->right=p;
}
void postorder(Node *h)
{
if(!flag||h==NULL)return;
postorder(h->left);
postorder(h->right);
if(h==root)printf("%d\n",h->value);
else printf("%d ",h->value);
}
void input()
{
scanf("%d",&n);
int i;
scanf("%d",&e[0]);
root=new Node(e[0]);
for(i=1;i<n;i++)//input
{
scanf("%d",&e[i]);
insert(root,e[i]);
}
id=0;
flag=true;
preorder(root);
if(flag)
{
printf("YES\n");
postorder(root);
return;
}
swap_Tree(root);//printf("sadasdjd %d\n",root->left->value);
id=0;
flag=true;
preorder(root);//printf("debug: %s\n",flag?"yes":"no");
if(!flag)
printf("NO\n");
else
{
printf("YES\n");
postorder(root);
}
}
int main()
{
input();
return 0;
}