RepeatN::fold

rust-itertools · Dec 15, 2023 · 7c5a5c0 · 7c5a5c0
1 parent 451c72a
commit 7c5a5c0
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/repeatn.rs b/src/repeatn.rs
@@ -44,6 +44,20 @@ where
     fn size_hint(&self) -> (usize, Option<usize>) {
         (self.n, Some(self.n))
     }
+
+    fn fold<B, F>(self, mut init: B, mut f: F) -> B
+    where
+        F: FnMut(B, Self::Item) -> B,
+    {
+        match self {
+            Self { elt: Some(elt), n } => {
+                debug_assert!(n > 0);
+                init = (1..n).map(|_| elt.clone()).fold(init, &mut f);
+                f(init, elt)
+            }
+            _ => init,
+        }
+    }
 }
 
 impl<A> DoubleEndedIterator for RepeatN<A>