題意:給一個二叉樹序列的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;
}