爆刷PAT(甲級)——之【1043】 Is It a Binary Search Tree(25 分)——BST二叉搜索樹定義

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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章