二叉查找树的建立与数的遍历。
AC代码:
#include<iostream>
#include<vector>
#include<map>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<list>
#include<set>
using namespace std;
vector<int> ori,pre1,pre2,post1,post2;
struct node
{
int data;
node* left;
node* right;
};
void insertt(node* &root,int data)
{
if(root==NULL)
{
root=new node;
root->data=data;
// cout<<data<<endl;
root->left=NULL;
root->right=NULL;
}
else
{
if(data<root->data)
{
insertt(root->left,data);
}
else
{
insertt(root->right,data);
}
}
}
void preorder(node* root)
{
if(root==NULL) return ;
pre1.push_back(root->data);
// cout<<pre1.back()<<endl;
preorder(root->left);
preorder(root->right);
}
void preorder1(node* root)
{
if(root==NULL) return ;
pre2.push_back(root->data);
preorder1(root->right);
preorder1(root->left);
}
void postorder(node* root)
{
if(root==NULL) return;
postorder(root->left);
postorder(root->right);
post1.push_back(root->data);
}
void postorder1(node* root)
{
if(root==NULL) return;
postorder1(root->right);
postorder1(root->left);
post2.push_back(root->data);
}
int main()
{
int n;
cin>>n;
node* rot=NULL;
for(int i=0;i<n;i++)
{
int d;
cin>>d;
ori.push_back(d);
insertt(rot,d);
}
preorder(rot);
preorder1(rot);
if(ori==pre1)
{
cout<<"YES"<<endl;
postorder(rot);
cout<<post1[0];
for(int i=1;i<post1.size();i++)
{
cout<<" "<<post1[i];
}
}
else if(ori==pre2)
{
cout<<"YES"<<endl;
postorder1(rot);
cout<<post2[0];
for(int i=1;i<post2.size();i++)
{
cout<<" "<<post2[i];
}
}
else
{
cout<<"NO"<<endl;
}
}