Specialize Powerset::nth

rust-itertools · May 2, 2024 · 125a19b · 125a19b
1 parent cd2a80b
commit 125a19b
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/powerset.rs b/src/powerset.rs
@@ -71,6 +71,20 @@ where
         }
     }
 
+    fn nth(&mut self, mut n: usize) -> Option<Self::Item> {
+        loop {
+            match self.combs.try_nth(n) {
+                Ok(item) => return Some(item),
+                Err(remaining) => {
+                    if !self.increment_k() {
+                        return None;
+                    }
+                    n = remaining;
+                }
+            }
+        }
+    }
+
     fn size_hint(&self) -> SizeHint {
         let k = self.combs.k();
         // Total bounds for source iterator.