chore(orama): updates random rank function for zip tree

oramasearch · Sep 3, 2023 · 3dc3c33 · 3dc3c33
1 parent 38cc867
commit 3dc3c33
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/packages/orama/src/trees/zip.ts b/packages/orama/src/trees/zip.ts
@@ -9,15 +9,9 @@ interface Node<V = unknown> {
   rank: number
 }
 
-function randomRank (maxAttempts = 32): number {
-  let heads = 0
-
-  for (let i = 0; i < maxAttempts; i++) {
-    if (Math.random() >= 0.5) break
-    heads += 1
-  }
-
-  return heads
+function randomRank(): number {
+  const r = Math.random()
+  return Math.floor(Math.log(1 - r) / Math.log(1 - 0.5))
 }
 
 export function create<V> (key: number, value: V): Node<V> {