add find

pastilhas · Oct 12, 2024 · b3a1ecc · b3a1ecc
1 parent 2daab3d
commit b3a1ecc
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/rbtree.v b/src/rbtree.v
@@ -131,7 +131,18 @@ fn (mut t RBTree[T]) delete[T](val T) bool {
 fn (mut t RBTree[T]) delete_fixup[T](node &Node[T]) {
 }
 
-fn (mut t RBTree[T]) find[T](val T) ?&Node[T] {
+fn (mut t RBTree[T]) find[T](val T) ?T {
+	for p := t.root; unsafe { p != 0 }; {
+		c := t.cmp(p.val, val)
+		if c == 0 {
+			return p.val
+		}
+		p = if c > 0 {
+			p.left
+		} else {
+			p.right
+		}
+	}
 	return none
 }