DS二叉排序树之创建和插入

DS二叉排序树之创建和插入

题目描述

给出一个数据序列，建立二叉排序树，并实现插入功能

对二叉排序树进行中序遍历，可以得到有序的数据序列

输入

第一行输入t，表示有t个数据序列

第二行输入n，表示首个序列包含n个数据

第三行输入n个数据，都是自然数且互不相同，数据之间用空格隔开

第四行输入m，表示要插入m个数据

从第五行起，输入m行，每行一个要插入的数据，都是自然数且和前面的数据不等

以此类推输入下一个示例

输出

第一行输出有序的数据序列，对二叉排序树进行中序遍历可以得到

从第二行起，输出插入第m个数据后的有序序列，输出m行

以此类推输出下一个示例的结果

样例输入

1
6
22 33 55 66 11 44
3
77
50
10

样例输出

11 22 33 44 55 66
11 22 33 44 55 66 77
11 22 33 44 50 55 66 77
10 11 22 33 44 50 55 66 77

#include <iostream>
using namespace std;

class BiNode{
    int data;
    BiNode *lChild;
    BiNode *rChild;
public:
    BiNode():lChild(NULL),rChild(NULL){}
    BiNode(int e):data(e),lChild(NULL),rChild(NULL){}
    friend class BiTree;
};

class BiTree{
    BiNode *root;
    void InsertNode(int data,BiNode *&r);
    void MidOrder(BiNode *t);
public:
    BiTree(int data){root = new BiNode(data);}
    void Insert(int key);
    void MidOrder();
};

void BiTree::InsertNode(int data,BiNode *&r) {
    if(data>r->data && r->rChild)
        InsertNode(data,r->rChild);
    else if(data < r->data && r->lChild)
        InsertNode(data,r->lChild);
    else if(data> r->data && !r->rChild)
    {
        BiNode *s = new BiNode(data);
        r->rChild = s;
    }
    else if(data<r->data && !r->lChild)
    {
        BiNode *s = new BiNode(data);
        r->lChild = s;
    }
}

void BiTree::Insert(int key) {
    InsertNode(key,root);
}

void BiTree::MidOrder(BiNode *t) {
    if(t)
    {
        MidOrder(t->lChild);
        cout<<t->data<<' ';
        MidOrder(t->rChild);
    }
}

void BiTree::MidOrder() {
    MidOrder(root);
    cout<<endl;
}

int main()
{
    int t;
    cin>>t;
    while (t--)
    {
        int n;
        cin>>n;
        int data;
        cin>>data;
        BiTree myTree(data);
        for(int i=1;i<n;i++)
        {
            cin>>data;
            myTree.Insert(data);
        }
        myTree.MidOrder();
        int m;
        cin>>m;
        while (m--)
        {
            cin>>data;
            myTree.Insert(data);
            myTree.MidOrder();
        }
    }
}