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

题目描述：

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

输入：

    输入第一行包括一个整数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 <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#define N 109

struct Node
{
    int key;
    int left;
    int right;
};

int tree[N];
Node treeNode[N];

void pre(int gen) {
    if(gen == -1) {
        return;
    }
    printf("%d ",treeNode[gen].key);
    pre(treeNode[gen].left);
    pre(treeNode[gen].right);
    
}

void middle(int gen) {
    if(gen == -1) {
        return;
    }
    middle(treeNode[gen].left);
    printf("%d ",treeNode[gen].key);
    middle(treeNode[gen].right);
    
}

void hou(int gen) {
    if(gen == -1) {
        return;
    }
    hou(treeNode[gen].left);
    hou(treeNode[gen].right);
    printf("%d ",treeNode[gen].key);
}
int main(int argc, char const *argv[])
{
    int n;
    while(scanf("%d",&n) != EOF) {
        for(int i = 0; i < n; i++) {
            scanf("%d",&tree[i]);
        }
        treeNode[0].key = tree[0];
        treeNode[0].left = -1;
        treeNode[0].right = -1;
        for(int i = 1; i < n; i++) {
            treeNode[i].key = tree[i];
            treeNode[i].left = -1;
            treeNode[i].right = -1;
            int lastTemp = -1;
            int temp = 0;
            bool dir = false;
            bool isEqual = false;
            while(temp != -1) {
                if(tree[i] < treeNode[temp].key) {
                    lastTemp = temp;
                    temp = treeNode[temp].left;
                    dir = false;
                }
                else if(tree[i] > treeNode[temp].key){
                    lastTemp = temp;
                    temp = treeNode[temp].right;
                    dir = true;
                }
                else {
                    isEqual = true;
                    break;
                }
            }
            if(isEqual) {
                continue;
            }
            if(dir == false) {
                treeNode[lastTemp].left = i;
            }
            else {
                treeNode[lastTemp].right = i;
            }
            
            
        }
        pre(0);
        printf("
");
        middle(0);
        printf("
");
        hou(0);
        printf("
");
    }
    return 0;
}