Merge pull request #5 from DanArmor/main

fix: return 1 for gcd_for_slice if all elements are 0.0
ichi-h · Oct 1, 2023 · 44feed1 · 44feed1
2 parents 2f03855 + 123f4e9
commit 44feed1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/util.rs b/src/util.rs
@@ -15,7 +15,13 @@ pub mod math {
         }
 
         let first = slice[0];
-        slice.iter().fold(
+        let mut iter = slice.iter().skip_while(|x| x == &&0);
+        let first = match iter.next() {
+            Some(v) => *v,
+            None => return 1
+        };
+
+        let gcd = iter.fold(
             first,
             |acc, cur| {
                 if *cur == 0 {
@@ -24,7 +30,8 @@ pub mod math {
                     gcd(*cur, acc)
                 }
             },
-        )
+        );
+        gcd
     }
 }