一种二叉树插入方法C++

#include <iostream>

// 定义二叉树节点结构
struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

// 二叉树插入方法
TreeNode* insertNode(TreeNode* root, int value) {
    // 如果根节点为空，则创建新节点作为根节点
    if (root == nullptr) {
        return new TreeNode(value);
    }

    // 如果插入值小于当前节点值，则插入到左子树
    if (value < root->val) {
        root->left = insertNode(root->left, value);
    }
    // 如果插入值大于等于当前节点值，则插入到右子树
    else {
        root->right = insertNode(root->right, value);
    }

    return root;
}

// 中序遍历二叉树（用于验证插入结果）
void inorderTraversal(TreeNode* root) {
    if (root == nullptr) {
        return;
    }

    inorderTraversal(root->left);
    std::cout << root->val << " ";
    inorderTraversal(root->right);
}

int main() {
    // 创建一个示例二叉树
    TreeNode* root = new TreeNode(4);
    root->left = new TreeNode(2);
    root->right = new TreeNode(7);
    root->left->left = new TreeNode(1);
    root->left->right = new TreeNode(3);

    std::cout << "原始二叉树中序遍历结果：";
    inorderTraversal(root);
    std::cout << std::endl;

    // 插入新节点
    int value = 5;
    root = insertNode(root, value);

    std::cout << "插入节点后的二叉树中序遍历结果：";
    inorderTraversal(root);
    std::cout << std::endl;

    return 0;
}