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 Snode
{
public:
    int data;
    Snode *lChild, *rChild;

    Snode() : lChild(NULL), rChild(NULL) {}
    Snode(int a) : data(a), lChild(NULL), rChild(NULL) {}
    ~Snode()
    {
        delete lChild;
        delete rChild;
    }
};

class Stree
{
    Snode *root;
    int time = 0;
    void create_tree(Snode *&t, int c)
    {
        if (t->data < c)
        {
            if (t->rChild == NULL)
            {
                Snode *a = new Snode(c);
                t->rChild = a;
                return;
            }
            else
                create_tree(t->rChild, c);
        }
        else
        {
            if (t->lChild == NULL)
            {
                Snode *a = new Snode(c);
                t->lChild = a;
                return;
            }
            else
                create_tree(t->lChild, c);
        }
    }
    void Mid_Order(Snode *t)
    {
        if (t)
        {
            Mid_Order(t->lChild);
            cout << t->data << " ";
            Mid_Order(t->rChild);
        }
    }

public:
    Stree() : root(NULL) {}
    void create_tree()
    {
        int c;
        cin >> c;
        if (root)
            create_tree(root, c);
        else
            root = new Snode(c);
    }
    void Mid_Order()
    {
        Mid_Order(root);
        cout << endl;
    }
    void insert(int t)
    {
        Snode *a = root;
        while (true)
        {
            if (a->data > t)
            {
                if (a->lChild == NULL)
                {
                    Snode *q = new Snode(t);
                    a->lChild = q;
                    return;
                }
                else
                    a = a->lChild;
            }
            else
            {
                if (a->rChild == NULL)
                {
                    Snode *q = new Snode(t);
                    a->rChild = q;
                    return;
                }
                else
                    a = a->rChild;
            }
        }
    }
    int search(int t)
    {

        Snode *a = root;
        while (a)
        {
            time++;
            if (t == a->data)
                return time;
            else if (t < a->data)
                a = a->lChild;
            else
                a = a->rChild;
        }
        return -1;
    }
    void set_time()
    {
        time=0;
    }
};
int main()
{
    int t;
    cin >> t;
    while (t--)
    {
        int n;
        cin >> n;
        Stree a;
        for (int i = 0; i < n; i++)
        {
            a.create_tree();
        }
        int nn;
        cin >> nn;
        int b;
        for (int i = 0; i < nn; i++)
        {
            
            cin >> b;
            cout << a.search(b) <<endl;
            a.set_time();
        }
    }
    return 0;
}