From 8f5a994802ebb616a10d99303214c11c0e81b290 Mon Sep 17 00:00:00 2001
From: jf-home <Orig5826@163.com>
Date: Mon, 24 Jun 2024 01:27:11 +0800
Subject: [PATCH] =?UTF-8?q?=E8=8E=AB=E5=90=8D=E5=85=B6=E5=A6=99=E8=B0=83?=
 =?UTF-8?q?=E8=AF=95=E9=80=9A=E8=BF=87=E4=BA=86=EF=BC=81WC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/tree.c | 30 +++++++++++++++++++++++++-----
 1 file changed, 25 insertions(+), 5 deletions(-)

diff --git a/src/tree.c b/src/tree.c
index 066d9d9..45d9bc6 100644
--- a/src/tree.c
+++ b/src/tree.c
@@ -1416,9 +1416,12 @@ static struct _tree_node* tree_turn_left(struct _tree* self, struct _tree_node*
 {
     assert(self != NULL);
     assert(root != NULL);
-    assert(root->right != NULL);
     struct _tree_node* node = root->right;
-    
+    if(node == NULL)
+    {
+        return root;
+    }
+
     if(root->parent != NULL)
     {
         if(root->parent->left == root)          // step1
@@ -1445,9 +1448,12 @@ static struct _tree_node* tree_turn_right(struct _tree* self, struct _tree_node*
 {
     assert(self != NULL);
     assert(root != NULL);
-    assert(root->left != NULL);
     struct _tree_node* node = root->left;
-    
+    if(node == NULL)
+    {
+        return root;
+    }
+
     if(root->parent != NULL)
     {
         if(root->parent->left == root)
@@ -1528,7 +1534,11 @@ static bool tree_avl_rebalance(struct _tree* self, struct _tree_node* root)
     {
         return true;
     }
-
+    if(root->left == NULL && root->right == NULL)
+    {
+        return true;
+    }
+    
     tree_set_balance(self, root);
     int balance = root->balance;
     if(balance > 1)
@@ -1725,6 +1735,10 @@ static bool tree_avl_delete_single_child(struct _tree* self, struct _tree_node*
                 node->right->parent = node->parent;
                 node->parent->left = node->right;
             }
+            else
+            {
+                node->parent->left = NULL;
+            }
         }
         else if(node->parent->right == node)
         {
@@ -1738,6 +1752,10 @@ static bool tree_avl_delete_single_child(struct _tree* self, struct _tree_node*
                 node->right->parent = node->parent;
                 node->parent->right = node->right;
             }
+            else
+            {
+                node->parent->right = NULL;
+            }
         }
 
         self->rebalance(self, node->parent);
@@ -1801,6 +1819,8 @@ bool tree_avl_delete(struct _tree* self, void* obj)
             {
                 node->parent->right = NULL;
             }
+
+            self->rebalance(self, node->parent);
         }
         tree_node_free(node);
     }