九度oj 题目1201：二叉排序树

题目1201：二叉排序树

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：7024

解决：2983

题目描述：

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

输入：

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

输出：

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

样例输入：

5
1 6 5 9 8

样例输出：

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

提示：

输入中可能有重复元素，但是输出的二叉树遍历序列中重复元素不用输出。

分析：重复元素不用输出，那么重复元素就不要放入二叉排序树中！否则输出的时候会很麻烦。

代码一：将中序遍历的元素放入一个容器中，最后在输出。可以控制格式(每行最后一个数据后没有空格）

#include <iostream>
#include <cstdlib>
#include <vector>
#include <cstdio>
using namespace std;

typedef struct node{
    int val;
    struct node *left;
    struct node *right;
} TreeNode, *Tree;

void insert(TreeNode *&root, int num){
    if(root == NULL){
        root = (TreeNode *)malloc(sizeof(TreeNode));
        //TreeNode *root = (TreeNode *)malloc(sizeof(TreeNode));这样写错误 
        root->val = num;
        root->left = NULL;
        root->right = NULL;
        return;
    }
    if(root->val == num)
        return;
    if(root->val > num)
        insert(root->left, num);
    else 
        insert(root->right, num);
}

void PreTraversal(Tree root, vector<int> &v){
    if(root == NULL)
        return;
    //cout << root->val << " ";
    v.push_back(root->val);
    PreTraversal(root->left, v);
    PreTraversal(root->right, v);
}

void InTraversal(Tree root, vector<int> &v){
    if(root == NULL)
        return;
    InTraversal(root->left, v);
    //cout << root->val << " ";
    v.push_back(root->val);
    InTraversal(root->right, v);
}

void PostTraversal(Tree root, vector<int> &v){
    if(root == NULL)
        return;
    PostTraversal(root->left, v);
    PostTraversal(root->right, v);
    //cout << root->val << " ";
    v.push_back(root->val);
}

void print(vector<int> v){
    int size = v.size();
    for(int i = 0; i < size; i++)
        printf("%d ", v[i]);
    printf("
");
}

int main(){
    vector<int> pre_v, in_v, post_v;
    int n, num;
    while(scanf("%d", &n) != EOF){
        pre_v.clear();
        in_v.clear();
        post_v.clear(); 
        Tree root = NULL;
        for(int i = 0; i < n; i++){
            scanf("%d", &num);
            insert(root, num);
        }
        PreTraversal(root, pre_v);
        InTraversal(root, in_v);
        PostTraversal(root, post_v);
        print(pre_v);
        print(in_v);
        print(post_v);
    }
    return 0;
}

代码二：直接遍历输出：

#include <iostream>
#include <cstdlib>
#include <cstdio>
using namespace std;

typedef struct node{
    int val;
    struct node *left;
    struct node *right;
} TreeNode, *Tree;

void insert(TreeNode *&root, int num){
    if(root == NULL){
        root = (TreeNode *)malloc(sizeof(TreeNode));
        //TreeNode *root = (TreeNode *)malloc(sizeof(TreeNode));这样写错误 
        root->val = num;
        root->left = NULL;
        root->right = NULL;
        return;
    }
    if(root->val == num)
        return;
    if(root->val > num)
        insert(root->left, num);
    else 
        insert(root->right, num);
}

void PreTraversal(Tree root){
    if(root == NULL)
        return;
    printf("%d ", root->val);
    PreTraversal(root->left);
    PreTraversal(root->right);
}

void InTraversal(Tree root){
    if(root == NULL)
        return;
    InTraversal(root->left);
    printf("%d ", root->val);
    InTraversal(root->right);
}

void PostTraversal(Tree root){
    if(root == NULL)
        return;
    PostTraversal(root->left);
    PostTraversal(root->right);
    printf("%d ", root->val);
}

int main(){
    int n, num;
    while(scanf("%d", &n) != EOF){ 
        Tree root = NULL;
        for(int i = 0; i < n; i++){
            scanf("%d", &num);
            insert(root, num);
        }
        PreTraversal(root);
        printf("
");
        InTraversal(root);
        printf("
");
        PostTraversal(root);
        printf("
");
    }
    return 0;
}