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

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

题目描述

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

在建立和插入操作后，都输出二叉树的先序遍历结果i

输入

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

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

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

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

输出

第一行输出一开始构建的二叉排序树的先序遍历结果

从第二行起，输出m行，每行输出插入一个数据到二叉排序树后的先序遍历结果

每行输出的遍历结果中，每个数据后面都带一个空格，最后一个数据也带。

输入样例1：

6
22 33 55 66 11 44
3
77
50
10

输出样例1：

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

输入样例2：

6
33 55 22 66 11 44
3
25
88
50

输出样例2：

33 22 11 55 44 66 
33 22 11 25 55 44 66 
33 22 11 25 55 44 66 88 
33 22 11 25 55 44 50 66 88 

参考代码：

#include <iostream>
#include <iomanip>
#include <vector>

using namespace std;

struct Node{
    int data;
    Node *left;
    Node *right;
};
Node *createNode(int data){
    Node *node = new Node;
    node->data = data;
    node->left = nullptr;
    node->right = nullptr;
    return node;
}
void insertNode(Node *root,int data){
    if(data < root->data){
        if(root->left == nullptr)
            root->left = createNode(data);
        else
            insertNode(root->left,data);
    }
    else{
        if(root->right == nullptr)
            root->right = createNode(data);
        else
            insertNode(root->right,data);
    }
}
void preOrder(Node *root){
    if(root == nullptr)
        return;
    cout << root->data << " ";
    preOrder(root->left);
    preOrder(root->right);
}
int main(){
    int n;
    cin >> n;
    vector<int> nums(n);
    for(int i = 0;i < n;i++){
        cin >> nums[i];
    }
    Node *root = createNode(nums[0]);
    for(int i = 1;i < n;i++){
        insertNode(root,nums[i]);
    }
    preOrder(root);
    cout << endl;
    int m;
    cin >> m;
    while (m--){
        int num;
        cin >> num;
        insertNode(root,num);
        preOrder(root);
        cout << endl;
    }
    return 0;
}