当先锋百科网

首页 1 2 3 4 5 6 7

二叉查找树的建立与数的遍历。

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;
    }

}