Fix the return type of binary search tree and avl tree

codes/c/chapter_tree/avl_tree.c

@@ -15,7 +15,7 @@ typedef struct avlTree avlTree;
 
 /* 构建 AVL 树 */
 avlTree *newAVLTree() {
-    avlTree *tree = (avlTree *) malloc(sizeof(avlTree));
+    avlTree *tree = (avlTree *)malloc(sizeof(avlTree));
     tree->root = NULL;
     return tree;
 }
@@ -132,11 +132,9 @@ TreeNode *insertHelper(TreeNode *node, int val) {
     return node;
 }
 
-
 /* 插入节点 */
-TreeNode *insert(avlTree *tree, int val) {
+void insert(avlTree *tree, int val) {
     tree->root = insertHelper(tree->root, val);
-    return tree->root;
 }
 
 /* 获取中序遍历中的下一个节点（仅适用于 root 有左子节点的情况） */
@@ -153,7 +151,7 @@ TreeNode *getInOrderNext(TreeNode *node) {
 
 /* 递归删除节点（辅助方法） */
 TreeNode *removeHelper(TreeNode *node, int val) {
-    TreeNode *child, *grandChild, *temp;
+    TreeNode *child, *grandChild;
     if (node == NULL) {
         return NULL;
     }
@@ -177,9 +175,13 @@ TreeNode *removeHelper(TreeNode *node, int val) {
             }
         } else {
             // 子节点数量 = 2 ，则将中序遍历的下个节点删除，并用该节点替换当前节点
-            temp = getInOrderNext(node->right);
+            TreeNode *temp = node->right;
+            while (temp->left != NULL) {
+                temp = temp->left;
+            }
+            int tempVal = temp->val;
             node->right = removeHelper(node->right, temp->val);
-            node->val = temp->val;
+            node->val = tempVal;
         }
     }
     // 更新节点高度
@@ -192,9 +194,8 @@ TreeNode *removeHelper(TreeNode *node, int val) {
 
 /* 删除节点 */
 // 由于引入了 stdio.h ，此处无法使用 remove 关键词
-TreeNode *removeNode(avlTree *tree, int val) {
+void removeNode(avlTree *tree, int val) {
     TreeNode *root = removeHelper(tree->root, val);
-    return root;
 }
 
 /* 查找节点 */
@@ -232,7 +233,7 @@ void testRemove(avlTree *tree, int val) {
 /* Driver Code */
 int main() {
     /* 初始化空 AVL 树 */
-    avlTree *tree = (avlTree *) newAVLTree();
+    avlTree *tree = (avlTree *)newAVLTree();
     /* 插入节点 */
     // 请关注插入节点后，AVL 树是如何保持平衡的
     testInsert(tree, 1);

codes/c/chapter_tree/binary_search_tree.c

@@ -69,15 +69,16 @@ TreeNode *search(binarySearchTree *bst, int num) {
 }
 
 /* 插入节点 */
-TreeNode *insert(binarySearchTree *bst, int num) {
+void insert(binarySearchTree *bst, int num) {
     // 若树为空，直接提前返回
-    if (bst->root == NULL) return NULL;
+    if (bst->root == NULL)
+        return;
     TreeNode *cur = bst->root, *pre = NULL;
     // 循环查找，越过叶节点后跳出
     while (cur != NULL) {
         // 找到重复节点，直接返回
         if (cur->val == num) {
-            return NULL;
+            return;
         }
         pre = cur;
         if (cur->val < num) {
@@ -95,24 +96,14 @@ TreeNode *insert(binarySearchTree *bst, int num) {
     } else {
         pre->left = node;
     }
-    return node;
-}
-
-/* 获取中序遍历中的下一个节点（仅适用于 root 有左子节点的情况） */
-TreeNode *getInOrderNext(TreeNode *root) {
-    if (root == NULL) return root;
-    // 循环访问左子节点，直到叶节点时为最小节点，跳出
-    while (root->left != NULL) {
-        root = root->left;
-    }
-    return root;
 }
 
 /* 删除节点 */
 // 由于引入了 stdio.h ，此处无法使用 remove 关键词
-TreeNode *removeNode(binarySearchTree *bst, int num) {
+void removeNode(binarySearchTree *bst, int num) {
     // 若树为空，直接提前返回
-    if (bst->root == NULL) return NULL;
+    if (bst->root == NULL)
+        return;
     TreeNode *cur = bst->root, *pre = NULL;
     // 循环查找，越过叶节点后跳出
     while (cur != NULL) {
@@ -128,9 +119,8 @@ TreeNode *removeNode(binarySearchTree *bst, int num) {
         }
     }
     // 若无待删除节点，则直接返回
-    if (cur == NULL) {
-        return NULL;
-    }
+    if (cur == NULL)
+        return;
     // 判断待删除节点是否存在子节点
     if (cur->left == NULL || cur->right == NULL) {
         /* 子节点数量 = 0 or 1 */
@@ -145,14 +135,16 @@ TreeNode *removeNode(binarySearchTree *bst, int num) {
     } else {
         /* 子节点数量 = 2 */
         // 获取中序遍历中 cur 的下一个节点
-        TreeNode *nex = getInOrderNext(cur->right);
-        int tmp = nex->val;
-        // 递归删除节点 nex
-        removeNode(bst, nex->val);
-        // 将 nex 的值复制给 cur
-        cur->val = tmp;
+        TreeNode *tmp = cur->right;
+        while (tmp->left != NULL) {
+            tmp = tmp->left;
+        }
+        int tmpVal = tmp->val;
+        // 递归删除节点 tmp
+        removeNode(bst, tmp->val);
+        // 用 tmp 覆盖 cur
+        cur->val = tmpVal;
     }
-    return cur;
 }
 
 /* Driver Code */