diff options
| -rw-r--r-- | autofeur_db/src/trie.rs | 13 |
1 files 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"); |