PAT程序设计考题——甲级1043（Is It a Binary Search Tree ） C++实现

点击打开pat链接

#include<iostream>
#include<math.h>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<string>
#include<vector>
using namespace std;
#define INF 100000000
#define maxn 100010

struct node{
 int num;
 node *lchild,*rchild;
};
node *newnode(int v){
 node *root=new node;
 root->num=v;
 root->lchild=root->rchild=NULL;
 return root;
}
void insert(node *&root,int v)
{
 if(root==NULL)
 {
  root=newnode(v);
  return;
  }
// if(root->num==v) return;注意这个条件是不需要的
 if(root->num>v)
 insert(root->lchild,v);
// if(root->num<v)
else
 insert(root->rchild,v);
}

int pre[maxn];
int post[maxn];
int m=0,n=0;
void pret(node *&root)
{
 if(root==NULL) return;
  pre[m++]=root->num;
 pret(root->lchild);
 pret(root->rchild);
}
int w=0;
void pret2(node *&root)
{
 if(root==NULL) return;
  pre[w++]=root->num;
   pret2(root->rchild);
 pret2(root->lchild);

}
bool issame(int a[],int b[],int num)
{  for(int i=0;i<num;i++)
{
 if(a[i]!=b[i]) return false;
}
 return true;
}
void postt(node *root)
{ if(root==NULL) return;
     postt(root->lchild);
 postt(root->rchild);
 post[n++]=root->num;
}
int q=0;
void postt2(node *root)
{ if(root==NULL) return;
postt2(root->rchild);
     postt2(root->lchild);
 
 post[q++]=root->num;
}
int main()
{   int num;
cin>>num;
node *root=NULL;//注意这边一定要为空否则会出错
int start[maxn];
for(int i=0;i<num;i++)
{
cin>>start[i];
insert(root,start[i]);
 }
 pret(root);
 if(issame(start,pre,num))
 {
  cout<<"YES"<<endl;
  postt(root);
  for(int i=0;i<num;i++){
 
  cout<<post[i];
  if(i<num-1) cout<<" ";}
  return 0;
 }
 pret2(root);
 if(issame(start,pre,num))
 {
  cout<<"YES"<<endl;
  postt2(root);
  for(int i=0;i<num;i++){
 
  cout<<post[i];
  if(i<num-1) cout<<" ";}
 }
 else cout<<"NO"<<endl;
 
 return 0;
}