Fix the return type of binary search tree and avl tree

codes/java/chapter_tree/avl_tree.java

@@ -92,9 +92,8 @@ class AVLTree {
     }
 
     /* 插入节点 */
-    public TreeNode insert(int val) {
+    public void insert(int val) {
         root = insertHelper(root, val);
-        return root;
     }
 
     /* 递归插入节点（辅助方法） */
@@ -116,9 +115,8 @@ class AVLTree {
     }
 
     /* 删除节点 */
-    public TreeNode remove(int val) {
+    public void remove(int val) {
         root = removeHelper(root, val);
-        return root;
     }
 
     /* 递归删除节点（辅助方法） */
@@ -141,7 +139,10 @@ class AVLTree {
                     node = child;
             } else {
                 // 子节点数量 = 2 ，则将中序遍历的下个节点删除，并用该节点替换当前节点
-                TreeNode temp = getInOrderNext(node.right);
+                TreeNode temp = node.right;
+                while (temp.left != null) {
+                    temp = temp.left;
+                }
                 node.right = removeHelper(node.right, temp.val);
                 node.val = temp.val;
             }
@@ -153,17 +154,6 @@ class AVLTree {
         return node;
     }
 
-    /* 获取中序遍历中的下一个节点（仅适用于 root 有左子节点的情况） */
-    private TreeNode getInOrderNext(TreeNode node) {
-        if (node == null)
-            return node;
-        // 循环访问左子节点，直到叶节点时为最小节点，跳出
-        while (node.left != null) {
-            node = node.left;
-        }
-        return node;
-    }
-
     /* 查找节点 */
     public TreeNode search(int val) {
         TreeNode cur = root;