Algorithms/red-black-tree-0.rs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed balancing algorithm bugs and extracted it to separate non-recursive function

Browse Source
This commit is contained in:
Andrew Golovashevich 2025-12-29 20:20:38 +03:00
parent f349a8751b
commit c1a9b8a6bc
1 changed files with 37 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
src/algo/add.rs
View File

@ -170,36 +170,49 @@ fn _add_recursive<Ctx: AddRecContext>(
}
}
if ctx.getColor(stack.getParent().node.clone()) == Color::BLACK {
return; // todo break
}
if !stack.hasGrandParent() {
ctx.setColor(stack.getParent().node.clone(), Color::BLACK);
return; // todo break
}
unsafe { balance(ctx, stack) }
}
let mut symCtx =
DynamicDirectedBinaryTreeContextFromContext::wrap_mut(ctx, stack.getGrandParent().dir);
let uncle = symCtx.getOppositeChild(stack.getGrandParent().node.clone());
if let Some(uncle) = uncle {
if ctx.getColor(uncle.clone()) == Color::RED {
unsafe fn balance<Ctx: AddRecContext>(ctx: &mut Ctx, stack: &mut Stack<Frame<Ctx::NodeRef>>) {
loop {
if ctx.getColor(stack.getParent().node.clone()) == Color::BLACK {
break;
}
if !stack.hasGrandParent() {
ctx.setColor(stack.getParent().node.clone(), Color::BLACK);
ctx.setColor(uncle, Color::BLACK);
ctx.setColor(stack.getGrandParent().node.clone(), Color::RED);
return; // todo current = grandparent
break;
}
}
match stack.getGrandParent().dir == stack.getParent().dir {
true => {
todo!("__rotateOpposite_Switch(grandParent)");
stack.replaceTop(stack.pop());
return; // todo continue
let mut symCtx =
DynamicDirectedBinaryTreeContextFromContext::wrap_mut(ctx, stack.getGrandParent().dir);
let uncle = symCtx.getOppositeChild(stack.getGrandParent().node.clone());
if let Some(uncle) = uncle {
if ctx.getColor(uncle.clone()) == Color::RED {
ctx.setColor(stack.getParent().node.clone(), Color::BLACK);
ctx.setColor(uncle, Color::BLACK);
ctx.setColor(stack.getGrandParent().node.clone(), Color::RED);
if !stack.hasGrandGrandParent() {
break;
}
stack.popDiscard();
stack.popDiscard();
continue;
}
}
false => {
if stack.getGrandParent().dir != stack.getParent().dir {
todo!("__rotateForward_Forward(parent, grandParent)");
stack.popDiscard();
return; // todo current = parent
stack.replaceTop(Frame {
node: stack.getParent().node,
dir: stack.getGrandParent().dir,
})
}
ctx.setColor(stack.getParent().node.clone(), Color::BLACK);
ctx.setColor(stack.getGrandParent().node.clone(), Color::RED);
todo!("__rotateOpposite_Switch(grandParent)");
break;
}
}