BST insert(root, value) 递归方法

Question

编辑：我已经想通了 - 放弃了这个设计并重新开始，它成功了！谢谢你的建议。

我正在做 BST 算法作业，但我绝望地坚持使用插入方法。我在网上找到的所有资源都有一个与我创建的类似的版本，但我没有通过我们教授给我们的 JUnit 测试。我可以传递基本情况（root.payload == 值的空根和二叉树）。不过，我似乎无法通过下一个测试。这是我的 insert(root, value) 方法的代码：

public static Node<Integer> insert(Node<Integer> root, Integer value) {
    if (root == null) {
        root = new Node<>(value);
    } else if (root.payload.equals(value)) {
        return root;
    } else if (value < root.payload) {
        return root.left = insert(root.left, value);
    } else {
        return root.right = insert(root.right, value);
    }
    return root;
}

最终返回的是 original 根节点，但据我了解，最后返回的值应该是我的新节点。我查看了我的教科书和一些在线资源，它们都与这个设计非常相似，所以我很困惑为什么它不起作用。我尝试了其他一些设计，但最终都是 NullPointerException。除了检查节点的有效负载之外，提供给我们的 JUnit 测试还验证我们是否将节点插入到了正确的位置。我没有通过这两项测试。这是我们使用递归的第一个任务，所以我还是很陌生。任何指导将不胜感激！

Answer 1

如果您希望方法的返回值是新插入的节点，则还必须在 elseIf 子句中添加 return 语句，以便在递归堆栈展开时返回新创建的节点。检查下面的代码

public static Node<Integer> insert(Node<Integer> root, Integer value) {
if (root == null) {
    root = new Node<>(value);
} else if (value <= root.payload) {
    return root.left = insert(root.left, value);   // add a return here
} else {
   return  root.right = insert(root.right, value);  // add a return here
}
return root;

}

Answer 2

public static Node<Integer> insert(Node<Integer> root, Integer value) {
    if (root == null) {
        root = new Node<>(value);
    } else if (value < root.payload) {
         root.left = insert(root.left, value);
    } else if (value > root.payload) {
         root.right = insert(root.right, value);
    }
    return root;
}

这将在正确的位置添加值并返回根。您可以打印树进行检查