From 7698cd57a9db15aaf5d2f13affb001b4832033a4 Mon Sep 17 00:00:00 2001
From: Matthieu Pignolet <matthieu@matthieu-dev.xyz>
Date: Thu, 6 Jun 2024 14:17:27 +0400
Subject: [PATCH] make weighted probabilities depending on node's children
 children count

---
 autofeur_db/src/trie.rs | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/autofeur_db/src/trie.rs b/autofeur_db/src/trie.rs
index a14c56e..1788d52 100644
--- a/autofeur_db/src/trie.rs
+++ b/autofeur_db/src/trie.rs
@@ -109,11 +109,14 @@ impl<'a> Trie<'a> {
         while current_node.child_nodes.len() != 0 {
             // We need to choose a random child based on weights
             let weighted = WeightedIndex::new(current_node.child_nodes.iter().map(|(_, node)| {
-                node.child_count / (node
-                    .child_nodes
-                    .iter()
-                    .map(|(_, b)| b.child_count)
-                    .sum::<u64>() + 1)
+                (node.child_count
+                    / (node
+                        .child_nodes
+                        .iter()
+                        .map(|(_, b)| b.child_count)
+                        .sum::<u64>()
+                        + 1)
+                    + 1)
             }))
             .expect("distribution creation should be valid");
 
-- 
2.39.5