From bd3258010a4128609bc85b0261352240fb42e72c Mon Sep 17 00:00:00 2001
From: wjf-hs <Orig5826@163.com>
Date: Tue, 25 Jun 2024 18:02:47 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BA=A2=E9=BB=91=E6=A0=91=E6=8F=92=E5=85=A5?=
 =?UTF-8?q?=E5=AE=9E=E7=8E=B0=E5=AE=8C=E6=88=90=EF=BC=8C=E5=A6=82=E6=9E=9C?=
 =?UTF-8?q?=E6=A3=80=E6=B5=8B=E5=85=B6=E6=AD=A3=E7=A1=AE=E6=80=A7=E5=91=A2?=
 =?UTF-8?q?=EF=BC=9F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/tree.c       | 2 ++
 test/test_tree.c | 4 ++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/tree.c b/src/tree.c
index 8ecf894..a82dd25 100644
--- a/src/tree.c
+++ b/src/tree.c
@@ -1124,6 +1124,7 @@ static bool tree_rb_rebalance(struct _tree* self, struct _tree_node* node)
                 }
                 grandfather->color = RBT_RED;
                 tree_turn_right(self, grandfather);
+                break;
             }
         }
         else
@@ -1149,6 +1150,7 @@ static bool tree_rb_rebalance(struct _tree* self, struct _tree_node* node)
                 }
                 grandfather->color = RBT_RED;
                 tree_turn_left(self, grandfather);
+                break;
             }
         }
     }
diff --git a/test/test_tree.c b/test/test_tree.c
index ced3a5a..288bb3d 100644
--- a/test/test_tree.c
+++ b/test/test_tree.c
@@ -249,6 +249,6 @@ void test_rbtree_num(void)
 
 void test_tree(void)
 {
-    test_avltree_num();
-    // test_rbtree_num();
+    // test_avltree_num();
+    test_rbtree_num();
 }