二叉树插入操作的 Java 实现与注意事项

聖光之護 2025-09-24 00:00:00 次阅读

本文将深入探讨如何在 java 中实现二叉树的插入操作，并提供一个优化后的迭代版本代码示例。理解二叉树插入的正确方法，以及避免在遍历过程中错误修改树结构，是编写高效、可靠的二叉树操作代码的关键。

首先，我们定义二叉树的节点和二叉树结构：

public static class Node {
    int data;
    Node left;
    Node right;

    Node(int d) {
        data = d;
        left = null;
        right = null;
    }
}

static class BTree {
    Node root;
}

接下来，我们来实现二叉树的插入方法。一个常见的错误是在插入过程中直接修改 t.root，这会导致树结构的混乱。正确的做法是使用一个临时的 Node 变量来遍历树，直到找到合适的插入位置。以下是一个使用迭代方式实现的 insert 方法：

static boolean insert(BTree t, int data) {
    Node newNode = new Node(data);
    if (t.root == null) {
        t.root = newNode;
        return true;
    } else {
        Node node = t.root;
        while (true) {
            if (node.data == data) {
                return false; // 节点已存在
            } else if (node.data > data) {
                if (node.left == null) {
                    node.left = newNode;
                    return true;
                }
                node = node.left;
            } else {
                if (node.right == null) {
                    node.right = newNode;
                    return true;
                }
                node = node.right;
            }
        }
    }
}

代码解释：

空树情况： 如果树为空（t.root == null），则将新节点设置为根节点，并返回 true。
非空树情况：
- 使用 node 变量从根节点开始遍历树。
- 使用 while(true) 循环，直到找到插入位置或发现节点已存在。
- 如果当前节点的值等于要插入的值，则说明节点已存在，返回 false。
- 如果当前节点的值大于要插入的值，则向左子树移动。如果左子树为空，则将新节点插入到左子树，并返回 true。
- 如果当前节点的值小于要插入的值，则向右子树移动。如果右子树为空，则将新节点插入到右子树，并返回 true。

注意事项：