【算法笔记第9.4节-二叉查找树BST 】问题 A: 二叉排序树

题目描述

输入一系列整数，建立二叉排序数，并进行前序，中序，后序遍历。

输入

输入第一行包括一个整数n(1<=n<=100)。接下来的一行包括n个整数。

输出

可能有多组测试数据，对于每组数据，将题目所给数据建立一个二叉排序树，并对二叉排序树进行前序、中序和后序遍历。每种遍历结果输出一行。每行最后一个数据之后有一个空格。

样例输入

1
2 
2
8 15 
4
21 10 5 39 

样例输出

2 
2 
2 
8 15 
8 15 
15 8 
21 10 5 39 
5 10 21 39 
5 10 39 21 

元素要去重。

 

#include<stdio.h>
#include<vector>
#include<algorithm>
#include<map>
using namespace std;
struct node
{
    int data;
    node *lchild, *rchild;
};
int n;
void insert(node *&root, int data)
{
    if(root==NULL)
    {
        root = new node;
        root->data = data;
        root->lchild = root->rchild = NULL;
        return;
    }
    if(data < root->data) insert(root->lchild, data);
    else insert(root->rchild, data);
}
int cnt = 0;
void pre(node *root)
{
    if(root==NULL)
        return;
    printf("%d",root->data);
    cnt++;
    if(cnt!=n)
        printf(" ");
    else
        printf("\n");
    pre(root->lchild);
    pre(root->rchild);

}

void in(node *root)
{
    if(root==NULL)
        return;
    in(root->lchild);
    printf("%d",root->data);
    cnt++;
    if(cnt!=n)
        printf(" ");
    else
        printf("\n");

    in(root->rchild);

}

void pos(node *root)
{
    if(root==NULL)
        return;
    pos(root->lchild);
    pos(root->rchild);
    printf("%d",root->data);
    cnt++;
    if(cnt!=n)
        printf(" ");
    else
        printf("\n");
}


int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        map<int, int> m;//要去重，这是个坑点
        node* root = NULL;
        int data, k = n;
        for(int i=0; i< k; i++)
        {
            scanf("%d", &data);
            if(m[data] == 0)
            {
                insert(root, data);
                m[data] = 1;
            }
            else
                n--;
        }
        cnt = 0;
        pre(root);
        cnt = 0;
        in(root);
        cnt = 0;
        pos(root);
    }
}