PAT 1043. Is It a Binary Search Tree (25)

37 篇文章 0 订阅
//写得太冗长,但好歹是过了,思路比较清晰~~~之前一直有一两个case不过
#include<iostream>
using namespace std;
#define MAX 1005
int prelist[MAX];
class Node
{
    public:
    int data;Node* left,*right;
    Node(int data)
    {
        this->data=data;
        this->left=this->right=NULL;
    }
};
bool isBST(int start, int n)
{
    if(n<=1)    return true;
    int elem=prelist[start];
    int i;
    for(i=1;i<n;i++)
    {
        if(prelist[start+i]>=elem)    break;
    }
    int left_len=i-1;
    for(;i<n;i++)
    {
        if(prelist[start+i]<elem)     return false;
    }
    if(isBST(start+1,left_len)&&isBST(start+left_len+1,n-left_len-1))
        return true;
    else
        return false;
}
bool ismBST(int start, int n)
{
    if(n<=1)    return true;
    int elem=prelist[start];
    int i;
    for(i=1;i<n;i++)
    {
        if(prelist[start+i]<elem)    break;
    }
    int left_len=i-1;
    for(;i<n;i++)
    {
        if(prelist[start+i]>=elem)     return false;
    }
    if(ismBST(start+1,left_len)&&ismBST(start+left_len+1,n-left_len-1))
        return true;
    else
        return false;
}
Node* build_BST(int start,int n)
{
    Node *p=NULL;
    if(n>0)
    {
        int elem=prelist[start];
        p=new Node(elem);
        int i=1;
        for(i=1;i<n;i++)
        {
            if(elem<=prelist[start+i])  break;
        }
        p->left=build_BST(start+1,i-1);
        p->right=build_BST(start+i,n-i);
    }
    return p;
}
Node* build_mBST(int start,int n)
{
    Node *p=NULL;
    if(n>0)
    {
        int elem=prelist[start];
        p=new Node(elem);
        int i=1;
        for(i=1;i<n;i++)
        {
            if(elem>prelist[start+i])  break;
        }
        p->left=build_mBST(start+1,i-1);
        p->right=build_mBST(start+i,n-i);
    }
    return p;
}
bool first=true;
void postorder(Node* p)
{
    if(p->left!=NULL)   postorder(p->left);
    if(p->right!=NULL) postorder(p->right);
    if(first)
    {
        first=false;cout<<p->data;
    }
    else
    {
        cout<<" "<<p->data;
    }
}
int main()
{
    int n;cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>prelist[i];
    }
    if( isBST(0,n))
    {
         cout<<"YES\n";
         Node *root=build_BST(0,n);
         postorder(root);
    }
   else if(ismBST(0,n))
   {
       cout<<"YES\n";
       Node *root=build_mBST(0,n);
       postorder(root);
   }
    else
    cout<<"NO\n";
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值