diff options
Diffstat (limited to 'src/solvers.rs')
-rw-r--r-- | src/solvers.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/solvers.rs b/src/solvers.rs index 0c01042..9bf0ad6 100644 --- a/src/solvers.rs +++ b/src/solvers.rs @@ -18,9 +18,9 @@ impl Wall { } pub fn calculate_row(&self, r: u32, stones: &mut [u32]) { - let mut len = 0; + let mut len = 1; let mut i = 0; - for &height in self.heights.iter() { + for &height in self.heights.iter().chain([r].iter()) { if height == r { stones[i] = len; i += 1; |