二叉排序樹（牛客北郵機試題）//構造二叉排序樹+先序中序後序遍歷

題目描述

輸入一系列整數，建立二叉排序樹，並進行前序，中序，後序遍歷。

輸入描述:

輸入第一行包括一個整數n(1<=n<=100)。
接下來的一行包括n個整數。

輸出描述:

可能有多組測試數據，對於每組數據，將題目所給數據建立一個二叉排序樹，並對二叉排序樹進行前序、中序和後序遍歷。
每種遍歷結果輸出一行。每行最後一個數據之後有一個空格。

輸入中可能有重複元素，但是輸出的二叉樹遍歷序列中重複元素不用輸出。

示例1

輸入

5
1 6 5 9 8

輸出

1 6 5 9 8 
1 5 6 8 9 
5 8 9 6 1 

AC代碼：

#include<bits/stdc++.h>
using namespace std;
struct Node
{
    int x;
    Node *l,*r;
};
void build(Node *t,int x)
{
    if(x<t->x)
    {
        if(t->l==NULL)
        {
            Node *n=new Node();
            n->x=x;
            n->l=NULL;
            n->r=NULL;
            t->l=n;
        }
        else build(t->l,x);
    }
    
    if(x>t->x)
    {
        if(t->r==NULL)
        {
            Node *n=new Node();
            n->x=x;
            n->l=NULL;
            n->r=NULL;
            t->r=n;
        }
        else build(t->r,x);
    }
    //cout<<1<<endl;
}
void pre(Node *n)
{
    printf("%d ",n->x);
    if(n->l!=NULL)pre(n->l);
    if(n->r!=NULL)pre(n->r);
}
void in(Node *n)
{
    
    if(n->l!=NULL)in(n->l);
    printf("%d ",n->x);
    if(n->r!=NULL)in(n->r);
}
void post(Node *n)
{
    
    if(n->l!=NULL)post(n->l);
    if(n->r!=NULL)post(n->r);
    printf("%d ",n->x);
}
int main()
{
    int n,i,j,x;
    while(~scanf("%d",&n))
    {
        Node *t;
        t=new Node();
        t->l=NULL;
        t->r=NULL;
        //memset(no,-1,sizeof(no));
        for(i=0;i<n;++i)
        {
            scanf("%d",&x);
            if(i==0)t->x=x;
            else
            {
                build(t,x);
            }
        }
        //cout<<1<<endl; 
        pre(t);
        printf("\n");
        in(t);
        printf("\n");
        post(t);
        printf("\n");
    }
}

優化後代碼：

#include<bits/stdc++.h>
using namespace std;
struct Node
{
    int x;
    Node *l,*r;
};
Node *build(Node *t,int x)
{
    if(t==NULL)
    {
        Node *n=new Node();
        n->x=x;
        n->l=n->r=NULL;
        return n;
    }
	if(x<t->x)
    {
        
        t->l=build(t->l,x);
    }else if(x>t->x)
    {
        t->r=build(t->r,x);
    }
    //cout<<1<<endl;
}
void pre(Node *n)
{
    printf("%d ",n->x);
    if(n->l!=NULL)pre(n->l);
    if(n->r!=NULL)pre(n->r);
}
void in(Node *n)
{
    
    if(n->l!=NULL)in(n->l);
    printf("%d ",n->x);
    if(n->r!=NULL)in(n->r);
}
void post(Node *n)
{
    
    if(n->l!=NULL)post(n->l);
    if(n->r!=NULL)post(n->r);
    printf("%d ",n->x);
}
int main()
{
    int n,i,j,x;
    while(~scanf("%d",&n))
    {
        Node *t=NULL;
        for(i=0;i<n;++i)
        {
            scanf("%d",&x);
            t=build(t,x);
        }
        //cout<<1<<endl; 
        pre(t);
        printf("\n");
        in(t);
        printf("\n");
        post(t);
        printf("\n");
    }
}