fixes

theuni · Jun 4, 2024 · 3e54b42 · 3e54b42
1 parent 2603320
commit 3e54b42
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/rbtree.cpp b/src/rbtree.cpp
@@ -70,10 +70,10 @@ class Tree
 
     template <int I>
     void fix_insert(Node* z) {
-        while (z != m_roots[I] && !z->m_positions[I].m_black) {
+        while (z != m_roots[I] && !z->m_positions[I].m_parent->m_positions[I].m_black) {
             if (z->m_positions[I].m_parent == z->m_positions[I].m_parent->m_positions[I].m_parent->m_positions[I].m_left) {
                 Node* y = z->m_positions[I].m_parent->m_positions[I].m_parent->m_positions[I].m_right;
-                if (y != nullptr && !z->m_positions[I].m_black) {
+                if (y != nullptr && !y->m_positions[I].m_black) {
                     z->m_positions[I].m_parent->m_positions[I].m_black = true;
                     y->m_positions[I].m_black = true;
                     z->m_positions[I].m_parent->m_positions[I].m_parent->m_positions[I].m_black = false;