From: Matthieu Pignolet <matthieu@matthieu-dev.xyz>
Date: Thu, 6 Jun 2024 10:13:41 +0000 (+0400)
Subject: make weighted probabilities depending on node's children children count
X-Git-Url: https://git.puffer.fish/?a=commitdiff_plain;h=e3f5437ed889dd7b736a891144b8123d2b92a81b;p=matthieu%2Fgru.git

make weighted probabilities depending on node's children children count
---

diff --git a/autofeur_db/src/trie.rs b/autofeur_db/src/trie.rs
index bbc2a93..6d0867a 100644
--- a/autofeur_db/src/trie.rs
+++ b/autofeur_db/src/trie.rs
@@ -108,12 +108,13 @@ impl<'a> Trie<'a> {
         let mut length = 0;
         while current_node.child_nodes.len() != 0 {
             // We need to choose a random child based on weights
-            let weighted = WeightedIndex::new(
-                current_node
+            let weighted = WeightedIndex::new(current_node.child_nodes.iter().map(|(_, node)| {
+                node.child_count / node
                     .child_nodes
                     .iter()
-                    .map(|(_, node)| node.child_count / (length + 1)),
-            )
+                    .map(|(_, b)| b.child_count)
+                    .sum::<u64>()
+            }))
             .expect("distribution creation should be valid");
 
             let (key, node) = current_node