From 05b5a4426c0eba6089bf23176aedf17fe1200e90 Mon Sep 17 00:00:00 2001 From: chriseth Date: Fri, 28 Oct 2022 17:28:35 +0200 Subject: [PATCH] Filter out zero-capacity nodes. --- src/graph/adjacencies.rs | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/graph/adjacencies.rs b/src/graph/adjacencies.rs index cc5041f..6e000d8 100644 --- a/src/graph/adjacencies.rs +++ b/src/graph/adjacencies.rs @@ -36,7 +36,10 @@ impl<'a> Adjacencies<'a> { *adjacencies.entry(node.clone()).or_default() += *c; } } - let mut result = adjacencies.into_iter().collect::>(); + let mut result = adjacencies + .into_iter() + .filter(|(_, cap)| *cap != U256::from(0)) + .collect::>(); result.sort_unstable_by_key(|(_, capacity)| Reverse(*capacity)); result }