itertools/
lib.rs

1#![warn(missing_docs, clippy::default_numeric_fallback)]
2#![crate_name = "itertools"]
3#![cfg_attr(not(feature = "use_std"), no_std)]
4
5//! Extra iterator adaptors, functions and macros.
6//!
7//! To extend [`Iterator`] with methods in this crate, import
8//! the [`Itertools`] trait:
9//!
10//! ```
11//! use itertools::Itertools;
12//! ```
13//!
14//! Now, new methods like [`interleave`](Itertools::interleave)
15//! are available on all iterators:
16//!
17//! ```
18//! use itertools::Itertools;
19//!
20//! let it = (1..3).interleave(vec![-1, -2]);
21//! itertools::assert_equal(it, vec![1, -1, 2, -2]);
22//! ```
23//!
24//! Most iterator methods are also provided as functions (with the benefit
25//! that they convert parameters using [`IntoIterator`]):
26//!
27//! ```
28//! use itertools::interleave;
29//!
30//! for elt in interleave(&[1, 2, 3], &[2, 3, 4]) {
31//!     /* loop body */
32//! }
33//! ```
34//!
35//! ## Crate Features
36//!
37//! - `use_std`
38//!   - Enabled by default.
39//!   - Disable to compile itertools using `#![no_std]`. This disables
40//!     any item that depend on allocations (see the `use_alloc` feature)
41//!     and hash maps (like `unique`, `counts`, `into_grouping_map` and more).
42//! - `use_alloc`
43//!   - Enabled by default.
44//!   - Enables any item that depend on allocations (like `chunk_by`,
45//!     `kmerge`, `join` and many more).
46//!
47//! ## Rust Version
48//!
49//! This version of itertools requires Rust 1.43.1 or later.
50
51#[cfg(not(feature = "use_std"))]
52extern crate core as std;
53
54#[cfg(feature = "use_alloc")]
55extern crate alloc;
56
57#[cfg(feature = "use_alloc")]
58use alloc::{collections::VecDeque, string::String, vec::Vec};
59
60pub use either::Either;
61
62use core::borrow::Borrow;
63use std::cmp::Ordering;
64#[cfg(feature = "use_std")]
65use std::collections::HashMap;
66#[cfg(feature = "use_std")]
67use std::collections::HashSet;
68use std::fmt;
69#[cfg(feature = "use_alloc")]
70use std::fmt::Write;
71#[cfg(feature = "use_std")]
72use std::hash::Hash;
73use std::iter::{once, IntoIterator};
74#[cfg(feature = "use_alloc")]
75type VecDequeIntoIter<T> = alloc::collections::vec_deque::IntoIter<T>;
76#[cfg(feature = "use_alloc")]
77type VecIntoIter<T> = alloc::vec::IntoIter<T>;
78use std::iter::FromIterator;
79
80#[macro_use]
81mod impl_macros;
82
83// for compatibility with no std and macros
84#[doc(hidden)]
85pub use std::iter as __std_iter;
86
87/// The concrete iterator types.
88pub mod structs {
89    #[cfg(feature = "use_alloc")]
90    pub use crate::adaptors::MultiProduct;
91    pub use crate::adaptors::{
92        Batching, Coalesce, Dedup, DedupBy, DedupByWithCount, DedupWithCount, FilterMapOk,
93        FilterOk, Interleave, InterleaveShortest, MapInto, MapOk, Positions, Product, PutBack,
94        TakeWhileRef, TupleCombinations, Update, WhileSome,
95    };
96    #[cfg(feature = "use_alloc")]
97    pub use crate::combinations::Combinations;
98    #[cfg(feature = "use_alloc")]
99    pub use crate::combinations_with_replacement::CombinationsWithReplacement;
100    pub use crate::cons_tuples_impl::ConsTuples;
101    #[cfg(feature = "use_std")]
102    pub use crate::duplicates_impl::{Duplicates, DuplicatesBy};
103    pub use crate::exactly_one_err::ExactlyOneError;
104    pub use crate::flatten_ok::FlattenOk;
105    pub use crate::format::{Format, FormatWith};
106    #[allow(deprecated)]
107    #[cfg(feature = "use_alloc")]
108    pub use crate::groupbylazy::GroupBy;
109    #[cfg(feature = "use_alloc")]
110    pub use crate::groupbylazy::{Chunk, ChunkBy, Chunks, Group, Groups, IntoChunks};
111    #[cfg(feature = "use_std")]
112    pub use crate::grouping_map::{GroupingMap, GroupingMapBy};
113    pub use crate::intersperse::{Intersperse, IntersperseWith};
114    #[cfg(feature = "use_alloc")]
115    pub use crate::kmerge_impl::{KMerge, KMergeBy};
116    pub use crate::merge_join::{Merge, MergeBy, MergeJoinBy};
117    #[cfg(feature = "use_alloc")]
118    pub use crate::multipeek_impl::MultiPeek;
119    pub use crate::pad_tail::PadUsing;
120    #[cfg(feature = "use_alloc")]
121    pub use crate::peek_nth::PeekNth;
122    pub use crate::peeking_take_while::PeekingTakeWhile;
123    #[cfg(feature = "use_alloc")]
124    pub use crate::permutations::Permutations;
125    #[cfg(feature = "use_alloc")]
126    pub use crate::powerset::Powerset;
127    pub use crate::process_results_impl::ProcessResults;
128    #[cfg(feature = "use_alloc")]
129    pub use crate::put_back_n_impl::PutBackN;
130    #[cfg(feature = "use_alloc")]
131    pub use crate::rciter_impl::RcIter;
132    pub use crate::repeatn::RepeatN;
133    #[allow(deprecated)]
134    pub use crate::sources::{Iterate, Unfold};
135    pub use crate::take_while_inclusive::TakeWhileInclusive;
136    #[cfg(feature = "use_alloc")]
137    pub use crate::tee::Tee;
138    pub use crate::tuple_impl::{CircularTupleWindows, TupleBuffer, TupleWindows, Tuples};
139    #[cfg(feature = "use_std")]
140    pub use crate::unique_impl::{Unique, UniqueBy};
141    pub use crate::with_position::WithPosition;
142    pub use crate::zip_eq_impl::ZipEq;
143    pub use crate::zip_longest::ZipLongest;
144    pub use crate::ziptuple::Zip;
145}
146
147/// Traits helpful for using certain `Itertools` methods in generic contexts.
148pub mod traits {
149    pub use crate::iter_index::IteratorIndex;
150    pub use crate::tuple_impl::HomogeneousTuple;
151}
152
153pub use crate::concat_impl::concat;
154pub use crate::cons_tuples_impl::cons_tuples;
155pub use crate::diff::diff_with;
156pub use crate::diff::Diff;
157#[cfg(feature = "use_alloc")]
158pub use crate::kmerge_impl::kmerge_by;
159pub use crate::minmax::MinMaxResult;
160pub use crate::peeking_take_while::PeekingNext;
161pub use crate::process_results_impl::process_results;
162pub use crate::repeatn::repeat_n;
163#[allow(deprecated)]
164pub use crate::sources::{iterate, unfold};
165#[allow(deprecated)]
166pub use crate::structs::*;
167pub use crate::unziptuple::{multiunzip, MultiUnzip};
168pub use crate::with_position::Position;
169pub use crate::ziptuple::multizip;
170mod adaptors;
171mod either_or_both;
172pub use crate::either_or_both::EitherOrBoth;
173#[doc(hidden)]
174pub mod free;
175#[doc(inline)]
176pub use crate::free::*;
177#[cfg(feature = "use_alloc")]
178mod combinations;
179#[cfg(feature = "use_alloc")]
180mod combinations_with_replacement;
181mod concat_impl;
182mod cons_tuples_impl;
183mod diff;
184#[cfg(feature = "use_std")]
185mod duplicates_impl;
186mod exactly_one_err;
187#[cfg(feature = "use_alloc")]
188mod extrema_set;
189mod flatten_ok;
190mod format;
191#[cfg(feature = "use_alloc")]
192mod group_map;
193#[cfg(feature = "use_alloc")]
194mod groupbylazy;
195#[cfg(feature = "use_std")]
196mod grouping_map;
197mod intersperse;
198mod iter_index;
199#[cfg(feature = "use_alloc")]
200mod k_smallest;
201#[cfg(feature = "use_alloc")]
202mod kmerge_impl;
203#[cfg(feature = "use_alloc")]
204mod lazy_buffer;
205mod merge_join;
206mod minmax;
207#[cfg(feature = "use_alloc")]
208mod multipeek_impl;
209mod pad_tail;
210#[cfg(feature = "use_alloc")]
211mod peek_nth;
212mod peeking_take_while;
213#[cfg(feature = "use_alloc")]
214mod permutations;
215#[cfg(feature = "use_alloc")]
216mod powerset;
217mod process_results_impl;
218#[cfg(feature = "use_alloc")]
219mod put_back_n_impl;
220#[cfg(feature = "use_alloc")]
221mod rciter_impl;
222mod repeatn;
223mod size_hint;
224mod sources;
225mod take_while_inclusive;
226#[cfg(feature = "use_alloc")]
227mod tee;
228mod tuple_impl;
229#[cfg(feature = "use_std")]
230mod unique_impl;
231mod unziptuple;
232mod with_position;
233mod zip_eq_impl;
234mod zip_longest;
235mod ziptuple;
236
237#[macro_export]
238/// Create an iterator over the “cartesian product” of iterators.
239///
240/// Iterator element type is like `(A, B, ..., E)` if formed
241/// from iterators `(I, J, ..., M)` with element types `I::Item = A`, `J::Item = B`, etc.
242///
243/// ```
244/// # use itertools::iproduct;
245/// #
246/// # fn main() {
247/// // Iterate over the coordinates of a 4 x 4 x 4 grid
248/// // from (0, 0, 0), (0, 0, 1), .., (0, 1, 0), (0, 1, 1), .. etc until (3, 3, 3)
249/// for (i, j, k) in iproduct!(0..4, 0..4, 0..4) {
250///    // ..
251/// }
252/// # }
253/// ```
254macro_rules! iproduct {
255    (@flatten $I:expr,) => (
256        $I
257    );
258    (@flatten $I:expr, $J:expr, $($K:expr,)*) => (
259        $crate::iproduct!(@flatten $crate::cons_tuples($crate::iproduct!($I, $J)), $($K,)*)
260    );
261    () => (
262        $crate::__std_iter::once(())
263    );
264    ($I:expr $(,)?) => (
265        $crate::__std_iter::IntoIterator::into_iter($I).map(|elt| (elt,))
266    );
267    ($I:expr, $J:expr $(,)?) => (
268        $crate::Itertools::cartesian_product(
269            $crate::__std_iter::IntoIterator::into_iter($I),
270            $crate::__std_iter::IntoIterator::into_iter($J),
271        )
272    );
273    ($I:expr, $J:expr, $($K:expr),+ $(,)?) => (
274        $crate::iproduct!(@flatten $crate::iproduct!($I, $J), $($K,)+)
275    );
276}
277
278#[macro_export]
279/// Create an iterator running multiple iterators in lockstep.
280///
281/// The `izip!` iterator yields elements until any subiterator
282/// returns `None`.
283///
284/// This is a version of the standard ``.zip()`` that's supporting more than
285/// two iterators. The iterator element type is a tuple with one element
286/// from each of the input iterators. Just like ``.zip()``, the iteration stops
287/// when the shortest of the inputs reaches its end.
288///
289/// **Note:** The result of this macro is in the general case an iterator
290/// composed of repeated `.zip()` and a `.map()`; it has an anonymous type.
291/// The special cases of one and two arguments produce the equivalent of
292/// `$a.into_iter()` and `$a.into_iter().zip($b)` respectively.
293///
294/// Prefer this macro `izip!()` over [`multizip`] for the performance benefits
295/// of using the standard library `.zip()`.
296///
297/// ```
298/// # use itertools::izip;
299/// #
300/// # fn main() {
301///
302/// // iterate over three sequences side-by-side
303/// let mut results = [0, 0, 0, 0];
304/// let inputs = [3, 7, 9, 6];
305///
306/// for (r, index, input) in izip!(&mut results, 0..10, &inputs) {
307///     *r = index * 10 + input;
308/// }
309///
310/// assert_eq!(results, [0 + 3, 10 + 7, 29, 36]);
311/// # }
312/// ```
313macro_rules! izip {
314    // @closure creates a tuple-flattening closure for .map() call. usage:
315    // @closure partial_pattern => partial_tuple , rest , of , iterators
316    // eg. izip!( @closure ((a, b), c) => (a, b, c) , dd , ee )
317    ( @closure $p:pat => $tup:expr ) => {
318        |$p| $tup
319    };
320
321    // The "b" identifier is a different identifier on each recursion level thanks to hygiene.
322    ( @closure $p:pat => ( $($tup:tt)* ) , $_iter:expr $( , $tail:expr )* ) => {
323        $crate::izip!(@closure ($p, b) => ( $($tup)*, b ) $( , $tail )*)
324    };
325
326    // unary
327    ($first:expr $(,)*) => {
328        $crate::__std_iter::IntoIterator::into_iter($first)
329    };
330
331    // binary
332    ($first:expr, $second:expr $(,)*) => {
333        $crate::izip!($first)
334            .zip($second)
335    };
336
337    // n-ary where n > 2
338    ( $first:expr $( , $rest:expr )* $(,)* ) => {
339        $crate::izip!($first)
340            $(
341                .zip($rest)
342            )*
343            .map(
344                $crate::izip!(@closure a => (a) $( , $rest )*)
345            )
346    };
347}
348
349#[macro_export]
350/// [Chain][`chain`] zero or more iterators together into one sequence.
351///
352/// The comma-separated arguments must implement [`IntoIterator`].
353/// The final argument may be followed by a trailing comma.
354///
355/// [`chain`]: Iterator::chain
356///
357/// # Examples
358///
359/// Empty invocations of `chain!` expand to an invocation of [`std::iter::empty`]:
360/// ```
361/// use std::iter;
362/// use itertools::chain;
363///
364/// let _: iter::Empty<()> = chain!();
365/// let _: iter::Empty<i8> = chain!();
366/// ```
367///
368/// Invocations of `chain!` with one argument expand to [`arg.into_iter()`](IntoIterator):
369/// ```
370/// use std::{ops::Range, slice};
371/// use itertools::chain;
372/// let _: <Range<_> as IntoIterator>::IntoIter = chain!((2..6),); // trailing comma optional!
373/// let _:     <&[_] as IntoIterator>::IntoIter = chain!(&[2, 3, 4]);
374/// ```
375///
376/// Invocations of `chain!` with multiple arguments [`.into_iter()`](IntoIterator) each
377/// argument, and then [`chain`] them together:
378/// ```
379/// use std::{iter::*, ops::Range, slice};
380/// use itertools::{assert_equal, chain};
381///
382/// // e.g., this:
383/// let with_macro:  Chain<Chain<Once<_>, Take<Repeat<_>>>, slice::Iter<_>> =
384///     chain![once(&0), repeat(&1).take(2), &[2, 3, 5],];
385///
386/// // ...is equivalent to this:
387/// let with_method: Chain<Chain<Once<_>, Take<Repeat<_>>>, slice::Iter<_>> =
388///     once(&0)
389///         .chain(repeat(&1).take(2))
390///         .chain(&[2, 3, 5]);
391///
392/// assert_equal(with_macro, with_method);
393/// ```
394macro_rules! chain {
395    () => {
396        core::iter::empty()
397    };
398    ($first:expr $(, $rest:expr )* $(,)?) => {
399        {
400            let iter = core::iter::IntoIterator::into_iter($first);
401            $(
402                let iter =
403                    core::iter::Iterator::chain(
404                        iter,
405                        core::iter::IntoIterator::into_iter($rest));
406            )*
407            iter
408        }
409    };
410}
411
412/// An [`Iterator`] blanket implementation that provides extra adaptors and
413/// methods.
414///
415/// This trait defines a number of methods. They are divided into two groups:
416///
417/// * *Adaptors* take an iterator and parameter as input, and return
418/// a new iterator value. These are listed first in the trait. An example
419/// of an adaptor is [`.interleave()`](Itertools::interleave)
420///
421/// * *Regular methods* are those that don't return iterators and instead
422/// return a regular value of some other kind.
423/// [`.next_tuple()`](Itertools::next_tuple) is an example and the first regular
424/// method in the list.
425pub trait Itertools: Iterator {
426    // adaptors
427
428    /// Alternate elements from two iterators until both have run out.
429    ///
430    /// Iterator element type is `Self::Item`.
431    ///
432    /// This iterator is *fused*.
433    ///
434    /// ```
435    /// use itertools::Itertools;
436    ///
437    /// let it = (1..7).interleave(vec![-1, -2]);
438    /// itertools::assert_equal(it, vec![1, -1, 2, -2, 3, 4, 5, 6]);
439    /// ```
440    fn interleave<J>(self, other: J) -> Interleave<Self, J::IntoIter>
441    where
442        J: IntoIterator<Item = Self::Item>,
443        Self: Sized,
444    {
445        interleave(self, other)
446    }
447
448    /// Alternate elements from two iterators until at least one of them has run
449    /// out.
450    ///
451    /// Iterator element type is `Self::Item`.
452    ///
453    /// ```
454    /// use itertools::Itertools;
455    ///
456    /// let it = (1..7).interleave_shortest(vec![-1, -2]);
457    /// itertools::assert_equal(it, vec![1, -1, 2, -2, 3]);
458    /// ```
459    fn interleave_shortest<J>(self, other: J) -> InterleaveShortest<Self, J::IntoIter>
460    where
461        J: IntoIterator<Item = Self::Item>,
462        Self: Sized,
463    {
464        adaptors::interleave_shortest(self, other.into_iter())
465    }
466
467    /// An iterator adaptor to insert a particular value
468    /// between each element of the adapted iterator.
469    ///
470    /// Iterator element type is `Self::Item`.
471    ///
472    /// This iterator is *fused*.
473    ///
474    /// ```
475    /// use itertools::Itertools;
476    ///
477    /// itertools::assert_equal((0..3).intersperse(8), vec![0, 8, 1, 8, 2]);
478    /// ```
479    fn intersperse(self, element: Self::Item) -> Intersperse<Self>
480    where
481        Self: Sized,
482        Self::Item: Clone,
483    {
484        intersperse::intersperse(self, element)
485    }
486
487    /// An iterator adaptor to insert a particular value created by a function
488    /// between each element of the adapted iterator.
489    ///
490    /// Iterator element type is `Self::Item`.
491    ///
492    /// This iterator is *fused*.
493    ///
494    /// ```
495    /// use itertools::Itertools;
496    ///
497    /// let mut i = 10;
498    /// itertools::assert_equal((0..3).intersperse_with(|| { i -= 1; i }), vec![0, 9, 1, 8, 2]);
499    /// assert_eq!(i, 8);
500    /// ```
501    fn intersperse_with<F>(self, element: F) -> IntersperseWith<Self, F>
502    where
503        Self: Sized,
504        F: FnMut() -> Self::Item,
505    {
506        intersperse::intersperse_with(self, element)
507    }
508
509    /// Returns an iterator over a subsection of the iterator.
510    ///
511    /// Works similarly to [`slice::get`](https://doc.rust-lang.org/std/primitive.slice.html#method.get).
512    ///
513    /// **Panics** for ranges `..=usize::MAX` and `0..=usize::MAX`.
514    ///
515    /// It's a generalisation of [`Iterator::take`] and [`Iterator::skip`],
516    /// and uses these under the hood.
517    /// Therefore, the resulting iterator is:
518    /// - [`ExactSizeIterator`] if the adapted iterator is [`ExactSizeIterator`].
519    /// - [`DoubleEndedIterator`] if the adapted iterator is [`DoubleEndedIterator`] and [`ExactSizeIterator`].
520    ///
521    /// # Unspecified Behavior
522    /// The result of indexing with an exhausted [`core::ops::RangeInclusive`] is unspecified.
523    ///
524    /// # Examples
525    ///
526    /// ```
527    /// use itertools::Itertools;
528    ///
529    /// let vec = vec![3, 1, 4, 1, 5];
530    ///
531    /// let mut range: Vec<_> =
532    ///         vec.iter().get(1..=3).copied().collect();
533    /// assert_eq!(&range, &[1, 4, 1]);
534    ///
535    /// // It works with other types of ranges, too
536    /// range = vec.iter().get(..2).copied().collect();
537    /// assert_eq!(&range, &[3, 1]);
538    ///
539    /// range = vec.iter().get(0..1).copied().collect();
540    /// assert_eq!(&range, &[3]);
541    ///
542    /// range = vec.iter().get(2..).copied().collect();
543    /// assert_eq!(&range, &[4, 1, 5]);
544    ///
545    /// range = vec.iter().get(..=2).copied().collect();
546    /// assert_eq!(&range, &[3, 1, 4]);
547    ///
548    /// range = vec.iter().get(..).copied().collect();
549    /// assert_eq!(range, vec);
550    /// ```
551    fn get<R>(self, index: R) -> R::Output
552    where
553        Self: Sized,
554        R: traits::IteratorIndex<Self>,
555    {
556        iter_index::get(self, index)
557    }
558
559    /// Create an iterator which iterates over both this and the specified
560    /// iterator simultaneously, yielding pairs of two optional elements.
561    ///
562    /// This iterator is *fused*.
563    ///
564    /// As long as neither input iterator is exhausted yet, it yields two values
565    /// via `EitherOrBoth::Both`.
566    ///
567    /// When the parameter iterator is exhausted, it only yields a value from the
568    /// `self` iterator via `EitherOrBoth::Left`.
569    ///
570    /// When the `self` iterator is exhausted, it only yields a value from the
571    /// parameter iterator via `EitherOrBoth::Right`.
572    ///
573    /// When both iterators return `None`, all further invocations of `.next()`
574    /// will return `None`.
575    ///
576    /// Iterator element type is
577    /// [`EitherOrBoth<Self::Item, J::Item>`](EitherOrBoth).
578    ///
579    /// ```rust
580    /// use itertools::EitherOrBoth::{Both, Right};
581    /// use itertools::Itertools;
582    /// let it = (0..1).zip_longest(1..3);
583    /// itertools::assert_equal(it, vec![Both(0, 1), Right(2)]);
584    /// ```
585    #[inline]
586    fn zip_longest<J>(self, other: J) -> ZipLongest<Self, J::IntoIter>
587    where
588        J: IntoIterator,
589        Self: Sized,
590    {
591        zip_longest::zip_longest(self, other.into_iter())
592    }
593
594    /// Create an iterator which iterates over both this and the specified
595    /// iterator simultaneously, yielding pairs of elements.
596    ///
597    /// **Panics** if the iterators reach an end and they are not of equal
598    /// lengths.
599    #[inline]
600    fn zip_eq<J>(self, other: J) -> ZipEq<Self, J::IntoIter>
601    where
602        J: IntoIterator,
603        Self: Sized,
604    {
605        zip_eq(self, other)
606    }
607
608    /// A “meta iterator adaptor”. Its closure receives a reference to the
609    /// iterator and may pick off as many elements as it likes, to produce the
610    /// next iterator element.
611    ///
612    /// Iterator element type is `B`.
613    ///
614    /// ```
615    /// use itertools::Itertools;
616    ///
617    /// // An adaptor that gathers elements in pairs
618    /// let pit = (0..4).batching(|it| {
619    ///            match it.next() {
620    ///                None => None,
621    ///                Some(x) => match it.next() {
622    ///                    None => None,
623    ///                    Some(y) => Some((x, y)),
624    ///                }
625    ///            }
626    ///        });
627    ///
628    /// itertools::assert_equal(pit, vec![(0, 1), (2, 3)]);
629    /// ```
630    ///
631    fn batching<B, F>(self, f: F) -> Batching<Self, F>
632    where
633        F: FnMut(&mut Self) -> Option<B>,
634        Self: Sized,
635    {
636        adaptors::batching(self, f)
637    }
638
639    /// Return an *iterable* that can group iterator elements.
640    /// Consecutive elements that map to the same key (“runs”), are assigned
641    /// to the same group.
642    ///
643    /// `ChunkBy` is the storage for the lazy grouping operation.
644    ///
645    /// If the groups are consumed in order, or if each group's iterator is
646    /// dropped without keeping it around, then `ChunkBy` uses no
647    /// allocations.  It needs allocations only if several group iterators
648    /// are alive at the same time.
649    ///
650    /// This type implements [`IntoIterator`] (it is **not** an iterator
651    /// itself), because the group iterators need to borrow from this
652    /// value. It should be stored in a local variable or temporary and
653    /// iterated.
654    ///
655    /// Iterator element type is `(K, Group)`: the group's key and the
656    /// group iterator.
657    ///
658    /// ```
659    /// use itertools::Itertools;
660    ///
661    /// // chunk data into runs of larger than zero or not.
662    /// let data = vec![1, 3, -2, -2, 1, 0, 1, 2];
663    /// // chunks:     |---->|------>|--------->|
664    ///
665    /// // Note: The `&` is significant here, `ChunkBy` is iterable
666    /// // only by reference. You can also call `.into_iter()` explicitly.
667    /// let mut data_grouped = Vec::new();
668    /// for (key, chunk) in &data.into_iter().chunk_by(|elt| *elt >= 0) {
669    ///     data_grouped.push((key, chunk.collect()));
670    /// }
671    /// assert_eq!(data_grouped, vec![(true, vec![1, 3]), (false, vec![-2, -2]), (true, vec![1, 0, 1, 2])]);
672    /// ```
673    #[cfg(feature = "use_alloc")]
674    fn chunk_by<K, F>(self, key: F) -> ChunkBy<K, Self, F>
675    where
676        Self: Sized,
677        F: FnMut(&Self::Item) -> K,
678        K: PartialEq,
679    {
680        groupbylazy::new(self, key)
681    }
682
683    /// See [`.chunk_by()`](Itertools::chunk_by).
684    #[deprecated(note = "Use .chunk_by() instead", since = "0.13.0")]
685    #[cfg(feature = "use_alloc")]
686    fn group_by<K, F>(self, key: F) -> ChunkBy<K, Self, F>
687    where
688        Self: Sized,
689        F: FnMut(&Self::Item) -> K,
690        K: PartialEq,
691    {
692        self.chunk_by(key)
693    }
694
695    /// Return an *iterable* that can chunk the iterator.
696    ///
697    /// Yield subiterators (chunks) that each yield a fixed number elements,
698    /// determined by `size`. The last chunk will be shorter if there aren't
699    /// enough elements.
700    ///
701    /// `IntoChunks` is based on `ChunkBy`: it is iterable (implements
702    /// `IntoIterator`, **not** `Iterator`), and it only buffers if several
703    /// chunk iterators are alive at the same time.
704    ///
705    /// Iterator element type is `Chunk`, each chunk's iterator.
706    ///
707    /// **Panics** if `size` is 0.
708    ///
709    /// ```
710    /// use itertools::Itertools;
711    ///
712    /// let data = vec![1, 1, 2, -2, 6, 0, 3, 1];
713    /// //chunk size=3 |------->|-------->|--->|
714    ///
715    /// // Note: The `&` is significant here, `IntoChunks` is iterable
716    /// // only by reference. You can also call `.into_iter()` explicitly.
717    /// for chunk in &data.into_iter().chunks(3) {
718    ///     // Check that the sum of each chunk is 4.
719    ///     assert_eq!(4, chunk.sum());
720    /// }
721    /// ```
722    #[cfg(feature = "use_alloc")]
723    fn chunks(self, size: usize) -> IntoChunks<Self>
724    where
725        Self: Sized,
726    {
727        assert!(size != 0);
728        groupbylazy::new_chunks(self, size)
729    }
730
731    /// Return an iterator over all contiguous windows producing tuples of
732    /// a specific size (up to 12).
733    ///
734    /// `tuple_windows` clones the iterator elements so that they can be
735    /// part of successive windows, this makes it most suited for iterators
736    /// of references and other values that are cheap to copy.
737    ///
738    /// ```
739    /// use itertools::Itertools;
740    /// let mut v = Vec::new();
741    ///
742    /// // pairwise iteration
743    /// for (a, b) in (1..5).tuple_windows() {
744    ///     v.push((a, b));
745    /// }
746    /// assert_eq!(v, vec![(1, 2), (2, 3), (3, 4)]);
747    ///
748    /// let mut it = (1..5).tuple_windows();
749    /// assert_eq!(Some((1, 2, 3)), it.next());
750    /// assert_eq!(Some((2, 3, 4)), it.next());
751    /// assert_eq!(None, it.next());
752    ///
753    /// // this requires a type hint
754    /// let it = (1..5).tuple_windows::<(_, _, _)>();
755    /// itertools::assert_equal(it, vec![(1, 2, 3), (2, 3, 4)]);
756    ///
757    /// // you can also specify the complete type
758    /// use itertools::TupleWindows;
759    /// use std::ops::Range;
760    ///
761    /// let it: TupleWindows<Range<u32>, (u32, u32, u32)> = (1..5).tuple_windows();
762    /// itertools::assert_equal(it, vec![(1, 2, 3), (2, 3, 4)]);
763    /// ```
764    fn tuple_windows<T>(self) -> TupleWindows<Self, T>
765    where
766        Self: Sized + Iterator<Item = T::Item>,
767        T: traits::HomogeneousTuple,
768        T::Item: Clone,
769    {
770        tuple_impl::tuple_windows(self)
771    }
772
773    /// Return an iterator over all windows, wrapping back to the first
774    /// elements when the window would otherwise exceed the length of the
775    /// iterator, producing tuples of a specific size (up to 12).
776    ///
777    /// `circular_tuple_windows` clones the iterator elements so that they can be
778    /// part of successive windows, this makes it most suited for iterators
779    /// of references and other values that are cheap to copy.
780    ///
781    /// ```
782    /// use itertools::Itertools;
783    /// let mut v = Vec::new();
784    /// for (a, b) in (1..5).circular_tuple_windows() {
785    ///     v.push((a, b));
786    /// }
787    /// assert_eq!(v, vec![(1, 2), (2, 3), (3, 4), (4, 1)]);
788    ///
789    /// let mut it = (1..5).circular_tuple_windows();
790    /// assert_eq!(Some((1, 2, 3)), it.next());
791    /// assert_eq!(Some((2, 3, 4)), it.next());
792    /// assert_eq!(Some((3, 4, 1)), it.next());
793    /// assert_eq!(Some((4, 1, 2)), it.next());
794    /// assert_eq!(None, it.next());
795    ///
796    /// // this requires a type hint
797    /// let it = (1..5).circular_tuple_windows::<(_, _, _)>();
798    /// itertools::assert_equal(it, vec![(1, 2, 3), (2, 3, 4), (3, 4, 1), (4, 1, 2)]);
799    /// ```
800    fn circular_tuple_windows<T>(self) -> CircularTupleWindows<Self, T>
801    where
802        Self: Sized + Clone + Iterator<Item = T::Item> + ExactSizeIterator,
803        T: tuple_impl::TupleCollect + Clone,
804        T::Item: Clone,
805    {
806        tuple_impl::circular_tuple_windows(self)
807    }
808    /// Return an iterator that groups the items in tuples of a specific size
809    /// (up to 12).
810    ///
811    /// See also the method [`.next_tuple()`](Itertools::next_tuple).
812    ///
813    /// ```
814    /// use itertools::Itertools;
815    /// let mut v = Vec::new();
816    /// for (a, b) in (1..5).tuples() {
817    ///     v.push((a, b));
818    /// }
819    /// assert_eq!(v, vec![(1, 2), (3, 4)]);
820    ///
821    /// let mut it = (1..7).tuples();
822    /// assert_eq!(Some((1, 2, 3)), it.next());
823    /// assert_eq!(Some((4, 5, 6)), it.next());
824    /// assert_eq!(None, it.next());
825    ///
826    /// // this requires a type hint
827    /// let it = (1..7).tuples::<(_, _, _)>();
828    /// itertools::assert_equal(it, vec![(1, 2, 3), (4, 5, 6)]);
829    ///
830    /// // you can also specify the complete type
831    /// use itertools::Tuples;
832    /// use std::ops::Range;
833    ///
834    /// let it: Tuples<Range<u32>, (u32, u32, u32)> = (1..7).tuples();
835    /// itertools::assert_equal(it, vec![(1, 2, 3), (4, 5, 6)]);
836    /// ```
837    ///
838    /// See also [`Tuples::into_buffer`].
839    fn tuples<T>(self) -> Tuples<Self, T>
840    where
841        Self: Sized + Iterator<Item = T::Item>,
842        T: traits::HomogeneousTuple,
843    {
844        tuple_impl::tuples(self)
845    }
846
847    /// Split into an iterator pair that both yield all elements from
848    /// the original iterator.
849    ///
850    /// **Note:** If the iterator is clonable, prefer using that instead
851    /// of using this method. Cloning is likely to be more efficient.
852    ///
853    /// Iterator element type is `Self::Item`.
854    ///
855    /// ```
856    /// use itertools::Itertools;
857    /// let xs = vec![0, 1, 2, 3];
858    ///
859    /// let (mut t1, t2) = xs.into_iter().tee();
860    /// itertools::assert_equal(t1.next(), Some(0));
861    /// itertools::assert_equal(t2, 0..4);
862    /// itertools::assert_equal(t1, 1..4);
863    /// ```
864    #[cfg(feature = "use_alloc")]
865    fn tee(self) -> (Tee<Self>, Tee<Self>)
866    where
867        Self: Sized,
868        Self::Item: Clone,
869    {
870        tee::new(self)
871    }
872
873    /// Convert each item of the iterator using the [`Into`] trait.
874    ///
875    /// ```rust
876    /// use itertools::Itertools;
877    ///
878    /// (1i32..42i32).map_into::<f64>().collect_vec();
879    /// ```
880    fn map_into<R>(self) -> MapInto<Self, R>
881    where
882        Self: Sized,
883        Self::Item: Into<R>,
884    {
885        adaptors::map_into(self)
886    }
887
888    /// Return an iterator adaptor that applies the provided closure
889    /// to every `Result::Ok` value. `Result::Err` values are
890    /// unchanged.
891    ///
892    /// ```
893    /// use itertools::Itertools;
894    ///
895    /// let input = vec![Ok(41), Err(false), Ok(11)];
896    /// let it = input.into_iter().map_ok(|i| i + 1);
897    /// itertools::assert_equal(it, vec![Ok(42), Err(false), Ok(12)]);
898    /// ```
899    fn map_ok<F, T, U, E>(self, f: F) -> MapOk<Self, F>
900    where
901        Self: Iterator<Item = Result<T, E>> + Sized,
902        F: FnMut(T) -> U,
903    {
904        adaptors::map_ok(self, f)
905    }
906
907    /// Return an iterator adaptor that filters every `Result::Ok`
908    /// value with the provided closure. `Result::Err` values are
909    /// unchanged.
910    ///
911    /// ```
912    /// use itertools::Itertools;
913    ///
914    /// let input = vec![Ok(22), Err(false), Ok(11)];
915    /// let it = input.into_iter().filter_ok(|&i| i > 20);
916    /// itertools::assert_equal(it, vec![Ok(22), Err(false)]);
917    /// ```
918    fn filter_ok<F, T, E>(self, f: F) -> FilterOk<Self, F>
919    where
920        Self: Iterator<Item = Result<T, E>> + Sized,
921        F: FnMut(&T) -> bool,
922    {
923        adaptors::filter_ok(self, f)
924    }
925
926    /// Return an iterator adaptor that filters and transforms every
927    /// `Result::Ok` value with the provided closure. `Result::Err`
928    /// values are unchanged.
929    ///
930    /// ```
931    /// use itertools::Itertools;
932    ///
933    /// let input = vec![Ok(22), Err(false), Ok(11)];
934    /// let it = input.into_iter().filter_map_ok(|i| if i > 20 { Some(i * 2) } else { None });
935    /// itertools::assert_equal(it, vec![Ok(44), Err(false)]);
936    /// ```
937    fn filter_map_ok<F, T, U, E>(self, f: F) -> FilterMapOk<Self, F>
938    where
939        Self: Iterator<Item = Result<T, E>> + Sized,
940        F: FnMut(T) -> Option<U>,
941    {
942        adaptors::filter_map_ok(self, f)
943    }
944
945    /// Return an iterator adaptor that flattens every `Result::Ok` value into
946    /// a series of `Result::Ok` values. `Result::Err` values are unchanged.
947    ///
948    /// This is useful when you have some common error type for your crate and
949    /// need to propagate it upwards, but the `Result::Ok` case needs to be flattened.
950    ///
951    /// ```
952    /// use itertools::Itertools;
953    ///
954    /// let input = vec![Ok(0..2), Err(false), Ok(2..4)];
955    /// let it = input.iter().cloned().flatten_ok();
956    /// itertools::assert_equal(it.clone(), vec![Ok(0), Ok(1), Err(false), Ok(2), Ok(3)]);
957    ///
958    /// // This can also be used to propagate errors when collecting.
959    /// let output_result: Result<Vec<i32>, bool> = it.collect();
960    /// assert_eq!(output_result, Err(false));
961    /// ```
962    fn flatten_ok<T, E>(self) -> FlattenOk<Self, T, E>
963    where
964        Self: Iterator<Item = Result<T, E>> + Sized,
965        T: IntoIterator,
966    {
967        flatten_ok::flatten_ok(self)
968    }
969
970    /// “Lift” a function of the values of the current iterator so as to process
971    /// an iterator of `Result` values instead.
972    ///
973    /// `processor` is a closure that receives an adapted version of the iterator
974    /// as the only argument — the adapted iterator produces elements of type `T`,
975    /// as long as the original iterator produces `Ok` values.
976    ///
977    /// If the original iterable produces an error at any point, the adapted
978    /// iterator ends and it will return the error iself.
979    ///
980    /// Otherwise, the return value from the closure is returned wrapped
981    /// inside `Ok`.
982    ///
983    /// # Example
984    ///
985    /// ```
986    /// use itertools::Itertools;
987    ///
988    /// type Item = Result<i32, &'static str>;
989    ///
990    /// let first_values: Vec<Item> = vec![Ok(1), Ok(0), Ok(3)];
991    /// let second_values: Vec<Item> = vec![Ok(2), Ok(1), Err("overflow")];
992    ///
993    /// // “Lift” the iterator .max() method to work on the Ok-values.
994    /// let first_max = first_values.into_iter().process_results(|iter| iter.max().unwrap_or(0));
995    /// let second_max = second_values.into_iter().process_results(|iter| iter.max().unwrap_or(0));
996    ///
997    /// assert_eq!(first_max, Ok(3));
998    /// assert!(second_max.is_err());
999    /// ```
1000    fn process_results<F, T, E, R>(self, processor: F) -> Result<R, E>
1001    where
1002        Self: Iterator<Item = Result<T, E>> + Sized,
1003        F: FnOnce(ProcessResults<Self, E>) -> R,
1004    {
1005        process_results(self, processor)
1006    }
1007
1008    /// Return an iterator adaptor that merges the two base iterators in
1009    /// ascending order.  If both base iterators are sorted (ascending), the
1010    /// result is sorted.
1011    ///
1012    /// Iterator element type is `Self::Item`.
1013    ///
1014    /// ```
1015    /// use itertools::Itertools;
1016    ///
1017    /// let a = (0..11).step_by(3);
1018    /// let b = (0..11).step_by(5);
1019    /// let it = a.merge(b);
1020    /// itertools::assert_equal(it, vec![0, 0, 3, 5, 6, 9, 10]);
1021    /// ```
1022    fn merge<J>(self, other: J) -> Merge<Self, J::IntoIter>
1023    where
1024        Self: Sized,
1025        Self::Item: PartialOrd,
1026        J: IntoIterator<Item = Self::Item>,
1027    {
1028        merge(self, other)
1029    }
1030
1031    /// Return an iterator adaptor that merges the two base iterators in order.
1032    /// This is much like [`.merge()`](Itertools::merge) but allows for a custom ordering.
1033    ///
1034    /// This can be especially useful for sequences of tuples.
1035    ///
1036    /// Iterator element type is `Self::Item`.
1037    ///
1038    /// ```
1039    /// use itertools::Itertools;
1040    ///
1041    /// let a = (0..).zip("bc".chars());
1042    /// let b = (0..).zip("ad".chars());
1043    /// let it = a.merge_by(b, |x, y| x.1 <= y.1);
1044    /// itertools::assert_equal(it, vec![(0, 'a'), (0, 'b'), (1, 'c'), (1, 'd')]);
1045    /// ```
1046
1047    fn merge_by<J, F>(self, other: J, is_first: F) -> MergeBy<Self, J::IntoIter, F>
1048    where
1049        Self: Sized,
1050        J: IntoIterator<Item = Self::Item>,
1051        F: FnMut(&Self::Item, &Self::Item) -> bool,
1052    {
1053        merge_join::merge_by_new(self, other, is_first)
1054    }
1055
1056    /// Create an iterator that merges items from both this and the specified
1057    /// iterator in ascending order.
1058    ///
1059    /// The function can either return an `Ordering` variant or a boolean.
1060    ///
1061    /// If `cmp_fn` returns `Ordering`,
1062    /// it chooses whether to pair elements based on the `Ordering` returned by the
1063    /// specified compare function. At any point, inspecting the tip of the
1064    /// iterators `I` and `J` as items `i` of type `I::Item` and `j` of type
1065    /// `J::Item` respectively, the resulting iterator will:
1066    ///
1067    /// - Emit `EitherOrBoth::Left(i)` when `i < j`,
1068    ///   and remove `i` from its source iterator
1069    /// - Emit `EitherOrBoth::Right(j)` when `i > j`,
1070    ///   and remove `j` from its source iterator
1071    /// - Emit `EitherOrBoth::Both(i, j)` when  `i == j`,
1072    ///   and remove both `i` and `j` from their respective source iterators
1073    ///
1074    /// ```
1075    /// use itertools::Itertools;
1076    /// use itertools::EitherOrBoth::{Left, Right, Both};
1077    ///
1078    /// let a = vec![0, 2, 4, 6, 1].into_iter();
1079    /// let b = (0..10).step_by(3);
1080    ///
1081    /// itertools::assert_equal(
1082    ///     a.merge_join_by(b, |i, j| i.cmp(j)),
1083    ///     vec![Both(0, 0), Left(2), Right(3), Left(4), Both(6, 6), Left(1), Right(9)]
1084    /// );
1085    /// ```
1086    ///
1087    /// If `cmp_fn` returns `bool`,
1088    /// it chooses whether to pair elements based on the boolean returned by the
1089    /// specified function. At any point, inspecting the tip of the
1090    /// iterators `I` and `J` as items `i` of type `I::Item` and `j` of type
1091    /// `J::Item` respectively, the resulting iterator will:
1092    ///
1093    /// - Emit `Either::Left(i)` when `true`,
1094    ///   and remove `i` from its source iterator
1095    /// - Emit `Either::Right(j)` when `false`,
1096    ///   and remove `j` from its source iterator
1097    ///
1098    /// It is similar to the `Ordering` case if the first argument is considered
1099    /// "less" than the second argument.
1100    ///
1101    /// ```
1102    /// use itertools::Itertools;
1103    /// use itertools::Either::{Left, Right};
1104    ///
1105    /// let a = vec![0, 2, 4, 6, 1].into_iter();
1106    /// let b = (0..10).step_by(3);
1107    ///
1108    /// itertools::assert_equal(
1109    ///     a.merge_join_by(b, |i, j| i <= j),
1110    ///     vec![Left(0), Right(0), Left(2), Right(3), Left(4), Left(6), Left(1), Right(6), Right(9)]
1111    /// );
1112    /// ```
1113    #[inline]
1114    fn merge_join_by<J, F, T>(self, other: J, cmp_fn: F) -> MergeJoinBy<Self, J::IntoIter, F>
1115    where
1116        J: IntoIterator,
1117        F: FnMut(&Self::Item, &J::Item) -> T,
1118        Self: Sized,
1119    {
1120        merge_join_by(self, other, cmp_fn)
1121    }
1122
1123    /// Return an iterator adaptor that flattens an iterator of iterators by
1124    /// merging them in ascending order.
1125    ///
1126    /// If all base iterators are sorted (ascending), the result is sorted.
1127    ///
1128    /// Iterator element type is `Self::Item`.
1129    ///
1130    /// ```
1131    /// use itertools::Itertools;
1132    ///
1133    /// let a = (0..6).step_by(3);
1134    /// let b = (1..6).step_by(3);
1135    /// let c = (2..6).step_by(3);
1136    /// let it = vec![a, b, c].into_iter().kmerge();
1137    /// itertools::assert_equal(it, vec![0, 1, 2, 3, 4, 5]);
1138    /// ```
1139    #[cfg(feature = "use_alloc")]
1140    fn kmerge(self) -> KMerge<<Self::Item as IntoIterator>::IntoIter>
1141    where
1142        Self: Sized,
1143        Self::Item: IntoIterator,
1144        <Self::Item as IntoIterator>::Item: PartialOrd,
1145    {
1146        kmerge(self)
1147    }
1148
1149    /// Return an iterator adaptor that flattens an iterator of iterators by
1150    /// merging them according to the given closure.
1151    ///
1152    /// The closure `first` is called with two elements *a*, *b* and should
1153    /// return `true` if *a* is ordered before *b*.
1154    ///
1155    /// If all base iterators are sorted according to `first`, the result is
1156    /// sorted.
1157    ///
1158    /// Iterator element type is `Self::Item`.
1159    ///
1160    /// ```
1161    /// use itertools::Itertools;
1162    ///
1163    /// let a = vec![-1f64, 2., 3., -5., 6., -7.];
1164    /// let b = vec![0., 2., -4.];
1165    /// let mut it = vec![a, b].into_iter().kmerge_by(|a, b| a.abs() < b.abs());
1166    /// assert_eq!(it.next(), Some(0.));
1167    /// assert_eq!(it.last(), Some(-7.));
1168    /// ```
1169    #[cfg(feature = "use_alloc")]
1170    fn kmerge_by<F>(self, first: F) -> KMergeBy<<Self::Item as IntoIterator>::IntoIter, F>
1171    where
1172        Self: Sized,
1173        Self::Item: IntoIterator,
1174        F: FnMut(&<Self::Item as IntoIterator>::Item, &<Self::Item as IntoIterator>::Item) -> bool,
1175    {
1176        kmerge_by(self, first)
1177    }
1178
1179    /// Return an iterator adaptor that iterates over the cartesian product of
1180    /// the element sets of two iterators `self` and `J`.
1181    ///
1182    /// Iterator element type is `(Self::Item, J::Item)`.
1183    ///
1184    /// ```
1185    /// use itertools::Itertools;
1186    ///
1187    /// let it = (0..2).cartesian_product("αβ".chars());
1188    /// itertools::assert_equal(it, vec![(0, 'α'), (0, 'β'), (1, 'α'), (1, 'β')]);
1189    /// ```
1190    fn cartesian_product<J>(self, other: J) -> Product<Self, J::IntoIter>
1191    where
1192        Self: Sized,
1193        Self::Item: Clone,
1194        J: IntoIterator,
1195        J::IntoIter: Clone,
1196    {
1197        adaptors::cartesian_product(self, other.into_iter())
1198    }
1199
1200    /// Return an iterator adaptor that iterates over the cartesian product of
1201    /// all subiterators returned by meta-iterator `self`.
1202    ///
1203    /// All provided iterators must yield the same `Item` type. To generate
1204    /// the product of iterators yielding multiple types, use the
1205    /// [`iproduct`] macro instead.
1206    ///
1207    /// The iterator element type is `Vec<T>`, where `T` is the iterator element
1208    /// of the subiterators.
1209    ///
1210    /// Note that the iterator is fused.
1211    ///
1212    /// ```
1213    /// use itertools::Itertools;
1214    /// let mut multi_prod = (0..3).map(|i| (i * 2)..(i * 2 + 2))
1215    ///     .multi_cartesian_product();
1216    /// assert_eq!(multi_prod.next(), Some(vec![0, 2, 4]));
1217    /// assert_eq!(multi_prod.next(), Some(vec![0, 2, 5]));
1218    /// assert_eq!(multi_prod.next(), Some(vec![0, 3, 4]));
1219    /// assert_eq!(multi_prod.next(), Some(vec![0, 3, 5]));
1220    /// assert_eq!(multi_prod.next(), Some(vec![1, 2, 4]));
1221    /// assert_eq!(multi_prod.next(), Some(vec![1, 2, 5]));
1222    /// assert_eq!(multi_prod.next(), Some(vec![1, 3, 4]));
1223    /// assert_eq!(multi_prod.next(), Some(vec![1, 3, 5]));
1224    /// assert_eq!(multi_prod.next(), None);
1225    /// ```
1226    ///
1227    /// If the adapted iterator is empty, the result is an iterator yielding a single empty vector.
1228    /// This is known as the [nullary cartesian product](https://en.wikipedia.org/wiki/Empty_product#Nullary_Cartesian_product).
1229    ///
1230    /// ```
1231    /// use itertools::Itertools;
1232    /// let mut nullary_cartesian_product = (0..0).map(|i| (i * 2)..(i * 2 + 2)).multi_cartesian_product();
1233    /// assert_eq!(nullary_cartesian_product.next(), Some(vec![]));
1234    /// assert_eq!(nullary_cartesian_product.next(), None);
1235    /// ```
1236    #[cfg(feature = "use_alloc")]
1237    fn multi_cartesian_product(self) -> MultiProduct<<Self::Item as IntoIterator>::IntoIter>
1238    where
1239        Self: Sized,
1240        Self::Item: IntoIterator,
1241        <Self::Item as IntoIterator>::IntoIter: Clone,
1242        <Self::Item as IntoIterator>::Item: Clone,
1243    {
1244        adaptors::multi_cartesian_product(self)
1245    }
1246
1247    /// Return an iterator adaptor that uses the passed-in closure to
1248    /// optionally merge together consecutive elements.
1249    ///
1250    /// The closure `f` is passed two elements, `previous` and `current` and may
1251    /// return either (1) `Ok(combined)` to merge the two values or
1252    /// (2) `Err((previous', current'))` to indicate they can't be merged.
1253    /// In (2), the value `previous'` is emitted by the iterator.
1254    /// Either (1) `combined` or (2) `current'` becomes the previous value
1255    /// when coalesce continues with the next pair of elements to merge. The
1256    /// value that remains at the end is also emitted by the iterator.
1257    ///
1258    /// Iterator element type is `Self::Item`.
1259    ///
1260    /// This iterator is *fused*.
1261    ///
1262    /// ```
1263    /// use itertools::Itertools;
1264    ///
1265    /// // sum same-sign runs together
1266    /// let data = vec![-1., -2., -3., 3., 1., 0., -1.];
1267    /// itertools::assert_equal(data.into_iter().coalesce(|x, y|
1268    ///         if (x >= 0.) == (y >= 0.) {
1269    ///             Ok(x + y)
1270    ///         } else {
1271    ///             Err((x, y))
1272    ///         }),
1273    ///         vec![-6., 4., -1.]);
1274    /// ```
1275    fn coalesce<F>(self, f: F) -> Coalesce<Self, F>
1276    where
1277        Self: Sized,
1278        F: FnMut(Self::Item, Self::Item) -> Result<Self::Item, (Self::Item, Self::Item)>,
1279    {
1280        adaptors::coalesce(self, f)
1281    }
1282
1283    /// Remove duplicates from sections of consecutive identical elements.
1284    /// If the iterator is sorted, all elements will be unique.
1285    ///
1286    /// Iterator element type is `Self::Item`.
1287    ///
1288    /// This iterator is *fused*.
1289    ///
1290    /// ```
1291    /// use itertools::Itertools;
1292    ///
1293    /// let data = vec![1., 1., 2., 3., 3., 2., 2.];
1294    /// itertools::assert_equal(data.into_iter().dedup(),
1295    ///                         vec![1., 2., 3., 2.]);
1296    /// ```
1297    fn dedup(self) -> Dedup<Self>
1298    where
1299        Self: Sized,
1300        Self::Item: PartialEq,
1301    {
1302        adaptors::dedup(self)
1303    }
1304
1305    /// Remove duplicates from sections of consecutive identical elements,
1306    /// determining equality using a comparison function.
1307    /// If the iterator is sorted, all elements will be unique.
1308    ///
1309    /// Iterator element type is `Self::Item`.
1310    ///
1311    /// This iterator is *fused*.
1312    ///
1313    /// ```
1314    /// use itertools::Itertools;
1315    ///
1316    /// let data = vec![(0, 1.), (1, 1.), (0, 2.), (0, 3.), (1, 3.), (1, 2.), (2, 2.)];
1317    /// itertools::assert_equal(data.into_iter().dedup_by(|x, y| x.1 == y.1),
1318    ///                         vec![(0, 1.), (0, 2.), (0, 3.), (1, 2.)]);
1319    /// ```
1320    fn dedup_by<Cmp>(self, cmp: Cmp) -> DedupBy<Self, Cmp>
1321    where
1322        Self: Sized,
1323        Cmp: FnMut(&Self::Item, &Self::Item) -> bool,
1324    {
1325        adaptors::dedup_by(self, cmp)
1326    }
1327
1328    /// Remove duplicates from sections of consecutive identical elements, while keeping a count of
1329    /// how many repeated elements were present.
1330    /// If the iterator is sorted, all elements will be unique.
1331    ///
1332    /// Iterator element type is `(usize, Self::Item)`.
1333    ///
1334    /// This iterator is *fused*.
1335    ///
1336    /// ```
1337    /// use itertools::Itertools;
1338    ///
1339    /// let data = vec!['a', 'a', 'b', 'c', 'c', 'b', 'b'];
1340    /// itertools::assert_equal(data.into_iter().dedup_with_count(),
1341    ///                         vec![(2, 'a'), (1, 'b'), (2, 'c'), (2, 'b')]);
1342    /// ```
1343    fn dedup_with_count(self) -> DedupWithCount<Self>
1344    where
1345        Self: Sized,
1346    {
1347        adaptors::dedup_with_count(self)
1348    }
1349
1350    /// Remove duplicates from sections of consecutive identical elements, while keeping a count of
1351    /// how many repeated elements were present.
1352    /// This will determine equality using a comparison function.
1353    /// If the iterator is sorted, all elements will be unique.
1354    ///
1355    /// Iterator element type is `(usize, Self::Item)`.
1356    ///
1357    /// This iterator is *fused*.
1358    ///
1359    /// ```
1360    /// use itertools::Itertools;
1361    ///
1362    /// let data = vec![(0, 'a'), (1, 'a'), (0, 'b'), (0, 'c'), (1, 'c'), (1, 'b'), (2, 'b')];
1363    /// itertools::assert_equal(data.into_iter().dedup_by_with_count(|x, y| x.1 == y.1),
1364    ///                         vec![(2, (0, 'a')), (1, (0, 'b')), (2, (0, 'c')), (2, (1, 'b'))]);
1365    /// ```
1366    fn dedup_by_with_count<Cmp>(self, cmp: Cmp) -> DedupByWithCount<Self, Cmp>
1367    where
1368        Self: Sized,
1369        Cmp: FnMut(&Self::Item, &Self::Item) -> bool,
1370    {
1371        adaptors::dedup_by_with_count(self, cmp)
1372    }
1373
1374    /// Return an iterator adaptor that produces elements that appear more than once during the
1375    /// iteration. Duplicates are detected using hash and equality.
1376    ///
1377    /// The iterator is stable, returning the duplicate items in the order in which they occur in
1378    /// the adapted iterator. Each duplicate item is returned exactly once. If an item appears more
1379    /// than twice, the second item is the item retained and the rest are discarded.
1380    ///
1381    /// ```
1382    /// use itertools::Itertools;
1383    ///
1384    /// let data = vec![10, 20, 30, 20, 40, 10, 50];
1385    /// itertools::assert_equal(data.into_iter().duplicates(),
1386    ///                         vec![20, 10]);
1387    /// ```
1388    #[cfg(feature = "use_std")]
1389    fn duplicates(self) -> Duplicates<Self>
1390    where
1391        Self: Sized,
1392        Self::Item: Eq + Hash,
1393    {
1394        duplicates_impl::duplicates(self)
1395    }
1396
1397    /// Return an iterator adaptor that produces elements that appear more than once during the
1398    /// iteration. Duplicates are detected using hash and equality.
1399    ///
1400    /// Duplicates are detected by comparing the key they map to with the keying function `f` by
1401    /// hash and equality. The keys are stored in a hash map in the iterator.
1402    ///
1403    /// The iterator is stable, returning the duplicate items in the order in which they occur in
1404    /// the adapted iterator. Each duplicate item is returned exactly once. If an item appears more
1405    /// than twice, the second item is the item retained and the rest are discarded.
1406    ///
1407    /// ```
1408    /// use itertools::Itertools;
1409    ///
1410    /// let data = vec!["a", "bb", "aa", "c", "ccc"];
1411    /// itertools::assert_equal(data.into_iter().duplicates_by(|s| s.len()),
1412    ///                         vec!["aa", "c"]);
1413    /// ```
1414    #[cfg(feature = "use_std")]
1415    fn duplicates_by<V, F>(self, f: F) -> DuplicatesBy<Self, V, F>
1416    where
1417        Self: Sized,
1418        V: Eq + Hash,
1419        F: FnMut(&Self::Item) -> V,
1420    {
1421        duplicates_impl::duplicates_by(self, f)
1422    }
1423
1424    /// Return an iterator adaptor that filters out elements that have
1425    /// already been produced once during the iteration. Duplicates
1426    /// are detected using hash and equality.
1427    ///
1428    /// Clones of visited elements are stored in a hash set in the
1429    /// iterator.
1430    ///
1431    /// The iterator is stable, returning the non-duplicate items in the order
1432    /// in which they occur in the adapted iterator. In a set of duplicate
1433    /// items, the first item encountered is the item retained.
1434    ///
1435    /// ```
1436    /// use itertools::Itertools;
1437    ///
1438    /// let data = vec![10, 20, 30, 20, 40, 10, 50];
1439    /// itertools::assert_equal(data.into_iter().unique(),
1440    ///                         vec![10, 20, 30, 40, 50]);
1441    /// ```
1442    #[cfg(feature = "use_std")]
1443    fn unique(self) -> Unique<Self>
1444    where
1445        Self: Sized,
1446        Self::Item: Clone + Eq + Hash,
1447    {
1448        unique_impl::unique(self)
1449    }
1450
1451    /// Return an iterator adaptor that filters out elements that have
1452    /// already been produced once during the iteration.
1453    ///
1454    /// Duplicates are detected by comparing the key they map to
1455    /// with the keying function `f` by hash and equality.
1456    /// The keys are stored in a hash set in the iterator.
1457    ///
1458    /// The iterator is stable, returning the non-duplicate items in the order
1459    /// in which they occur in the adapted iterator. In a set of duplicate
1460    /// items, the first item encountered is the item retained.
1461    ///
1462    /// ```
1463    /// use itertools::Itertools;
1464    ///
1465    /// let data = vec!["a", "bb", "aa", "c", "ccc"];
1466    /// itertools::assert_equal(data.into_iter().unique_by(|s| s.len()),
1467    ///                         vec!["a", "bb", "ccc"]);
1468    /// ```
1469    #[cfg(feature = "use_std")]
1470    fn unique_by<V, F>(self, f: F) -> UniqueBy<Self, V, F>
1471    where
1472        Self: Sized,
1473        V: Eq + Hash,
1474        F: FnMut(&Self::Item) -> V,
1475    {
1476        unique_impl::unique_by(self, f)
1477    }
1478
1479    /// Return an iterator adaptor that borrows from this iterator and
1480    /// takes items while the closure `accept` returns `true`.
1481    ///
1482    /// This adaptor can only be used on iterators that implement `PeekingNext`
1483    /// like `.peekable()`, `put_back` and a few other collection iterators.
1484    ///
1485    /// The last and rejected element (first `false`) is still available when
1486    /// `peeking_take_while` is done.
1487    ///
1488    ///
1489    /// See also [`.take_while_ref()`](Itertools::take_while_ref)
1490    /// which is a similar adaptor.
1491    fn peeking_take_while<F>(&mut self, accept: F) -> PeekingTakeWhile<Self, F>
1492    where
1493        Self: Sized + PeekingNext,
1494        F: FnMut(&Self::Item) -> bool,
1495    {
1496        peeking_take_while::peeking_take_while(self, accept)
1497    }
1498
1499    /// Return an iterator adaptor that borrows from a `Clone`-able iterator
1500    /// to only pick off elements while the predicate `accept` returns `true`.
1501    ///
1502    /// It uses the `Clone` trait to restore the original iterator so that the
1503    /// last and rejected element (first `false`) is still available when
1504    /// `take_while_ref` is done.
1505    ///
1506    /// ```
1507    /// use itertools::Itertools;
1508    ///
1509    /// let mut hexadecimals = "0123456789abcdef".chars();
1510    ///
1511    /// let decimals = hexadecimals.take_while_ref(|c| c.is_numeric())
1512    ///                            .collect::<String>();
1513    /// assert_eq!(decimals, "0123456789");
1514    /// assert_eq!(hexadecimals.next(), Some('a'));
1515    ///
1516    /// ```
1517    fn take_while_ref<F>(&mut self, accept: F) -> TakeWhileRef<Self, F>
1518    where
1519        Self: Clone,
1520        F: FnMut(&Self::Item) -> bool,
1521    {
1522        adaptors::take_while_ref(self, accept)
1523    }
1524
1525    /// Returns an iterator adaptor that consumes elements while the given
1526    /// predicate is `true`, *including* the element for which the predicate
1527    /// first returned `false`.
1528    ///
1529    /// The [`.take_while()`][std::iter::Iterator::take_while] adaptor is useful
1530    /// when you want items satisfying a predicate, but to know when to stop
1531    /// taking elements, we have to consume that first element that doesn't
1532    /// satisfy the predicate. This adaptor includes that element where
1533    /// [`.take_while()`][std::iter::Iterator::take_while] would drop it.
1534    ///
1535    /// The [`.take_while_ref()`][crate::Itertools::take_while_ref] adaptor
1536    /// serves a similar purpose, but this adaptor doesn't require [`Clone`]ing
1537    /// the underlying elements.
1538    ///
1539    /// ```rust
1540    /// # use itertools::Itertools;
1541    /// let items = vec![1, 2, 3, 4, 5];
1542    /// let filtered: Vec<_> = items
1543    ///     .into_iter()
1544    ///     .take_while_inclusive(|&n| n % 3 != 0)
1545    ///     .collect();
1546    ///
1547    /// assert_eq!(filtered, vec![1, 2, 3]);
1548    /// ```
1549    ///
1550    /// ```rust
1551    /// # use itertools::Itertools;
1552    /// let items = vec![1, 2, 3, 4, 5];
1553    ///
1554    /// let take_while_inclusive_result: Vec<_> = items
1555    ///     .iter()
1556    ///     .copied()
1557    ///     .take_while_inclusive(|&n| n % 3 != 0)
1558    ///     .collect();
1559    /// let take_while_result: Vec<_> = items
1560    ///     .into_iter()
1561    ///     .take_while(|&n| n % 3 != 0)
1562    ///     .collect();
1563    ///
1564    /// assert_eq!(take_while_inclusive_result, vec![1, 2, 3]);
1565    /// assert_eq!(take_while_result, vec![1, 2]);
1566    /// // both iterators have the same items remaining at this point---the 3
1567    /// // is lost from the `take_while` vec
1568    /// ```
1569    ///
1570    /// ```rust
1571    /// # use itertools::Itertools;
1572    /// #[derive(Debug, PartialEq)]
1573    /// struct NoCloneImpl(i32);
1574    ///
1575    /// let non_clonable_items: Vec<_> = vec![1, 2, 3, 4, 5]
1576    ///     .into_iter()
1577    ///     .map(NoCloneImpl)
1578    ///     .collect();
1579    /// let filtered: Vec<_> = non_clonable_items
1580    ///     .into_iter()
1581    ///     .take_while_inclusive(|n| n.0 % 3 != 0)
1582    ///     .collect();
1583    /// let expected: Vec<_> = vec![1, 2, 3].into_iter().map(NoCloneImpl).collect();
1584    /// assert_eq!(filtered, expected);
1585    fn take_while_inclusive<F>(self, accept: F) -> TakeWhileInclusive<Self, F>
1586    where
1587        Self: Sized,
1588        F: FnMut(&Self::Item) -> bool,
1589    {
1590        take_while_inclusive::TakeWhileInclusive::new(self, accept)
1591    }
1592
1593    /// Return an iterator adaptor that filters `Option<A>` iterator elements
1594    /// and produces `A`. Stops on the first `None` encountered.
1595    ///
1596    /// Iterator element type is `A`, the unwrapped element.
1597    ///
1598    /// ```
1599    /// use itertools::Itertools;
1600    ///
1601    /// // List all hexadecimal digits
1602    /// itertools::assert_equal(
1603    ///     (0..).map(|i| std::char::from_digit(i, 16)).while_some(),
1604    ///     "0123456789abcdef".chars());
1605    ///
1606    /// ```
1607    fn while_some<A>(self) -> WhileSome<Self>
1608    where
1609        Self: Sized + Iterator<Item = Option<A>>,
1610    {
1611        adaptors::while_some(self)
1612    }
1613
1614    /// Return an iterator adaptor that iterates over the combinations of the
1615    /// elements from an iterator.
1616    ///
1617    /// Iterator element can be any homogeneous tuple of type `Self::Item` with
1618    /// size up to 12.
1619    ///
1620    /// # Guarantees
1621    ///
1622    /// If the adapted iterator is deterministic,
1623    /// this iterator adapter yields items in a reliable order.
1624    ///
1625    /// ```
1626    /// use itertools::Itertools;
1627    ///
1628    /// let mut v = Vec::new();
1629    /// for (a, b) in (1..5).tuple_combinations() {
1630    ///     v.push((a, b));
1631    /// }
1632    /// assert_eq!(v, vec![(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]);
1633    ///
1634    /// let mut it = (1..5).tuple_combinations();
1635    /// assert_eq!(Some((1, 2, 3)), it.next());
1636    /// assert_eq!(Some((1, 2, 4)), it.next());
1637    /// assert_eq!(Some((1, 3, 4)), it.next());
1638    /// assert_eq!(Some((2, 3, 4)), it.next());
1639    /// assert_eq!(None, it.next());
1640    ///
1641    /// // this requires a type hint
1642    /// let it = (1..5).tuple_combinations::<(_, _, _)>();
1643    /// itertools::assert_equal(it, vec![(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]);
1644    ///
1645    /// // you can also specify the complete type
1646    /// use itertools::TupleCombinations;
1647    /// use std::ops::Range;
1648    ///
1649    /// let it: TupleCombinations<Range<u32>, (u32, u32, u32)> = (1..5).tuple_combinations();
1650    /// itertools::assert_equal(it, vec![(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]);
1651    /// ```
1652    fn tuple_combinations<T>(self) -> TupleCombinations<Self, T>
1653    where
1654        Self: Sized + Clone,
1655        Self::Item: Clone,
1656        T: adaptors::HasCombination<Self>,
1657    {
1658        adaptors::tuple_combinations(self)
1659    }
1660
1661    /// Return an iterator adaptor that iterates over the `k`-length combinations of
1662    /// the elements from an iterator.
1663    ///
1664    /// Iterator element type is `Vec<Self::Item>`. The iterator produces a new `Vec` per iteration,
1665    /// and clones the iterator elements.
1666    ///
1667    /// # Guarantees
1668    ///
1669    /// If the adapted iterator is deterministic,
1670    /// this iterator adapter yields items in a reliable order.
1671    ///
1672    /// ```
1673    /// use itertools::Itertools;
1674    ///
1675    /// let it = (1..5).combinations(3);
1676    /// itertools::assert_equal(it, vec![
1677    ///     vec![1, 2, 3],
1678    ///     vec![1, 2, 4],
1679    ///     vec![1, 3, 4],
1680    ///     vec![2, 3, 4],
1681    /// ]);
1682    /// ```
1683    ///
1684    /// Note: Combinations does not take into account the equality of the iterated values.
1685    /// ```
1686    /// use itertools::Itertools;
1687    ///
1688    /// let it = vec![1, 2, 2].into_iter().combinations(2);
1689    /// itertools::assert_equal(it, vec![
1690    ///     vec![1, 2], // Note: these are the same
1691    ///     vec![1, 2], // Note: these are the same
1692    ///     vec![2, 2],
1693    /// ]);
1694    /// ```
1695    #[cfg(feature = "use_alloc")]
1696    fn combinations(self, k: usize) -> Combinations<Self>
1697    where
1698        Self: Sized,
1699        Self::Item: Clone,
1700    {
1701        combinations::combinations(self, k)
1702    }
1703
1704    /// Return an iterator that iterates over the `k`-length combinations of
1705    /// the elements from an iterator, with replacement.
1706    ///
1707    /// Iterator element type is `Vec<Self::Item>`. The iterator produces a new `Vec` per iteration,
1708    /// and clones the iterator elements.
1709    ///
1710    /// ```
1711    /// use itertools::Itertools;
1712    ///
1713    /// let it = (1..4).combinations_with_replacement(2);
1714    /// itertools::assert_equal(it, vec![
1715    ///     vec![1, 1],
1716    ///     vec![1, 2],
1717    ///     vec![1, 3],
1718    ///     vec![2, 2],
1719    ///     vec![2, 3],
1720    ///     vec![3, 3],
1721    /// ]);
1722    /// ```
1723    #[cfg(feature = "use_alloc")]
1724    fn combinations_with_replacement(self, k: usize) -> CombinationsWithReplacement<Self>
1725    where
1726        Self: Sized,
1727        Self::Item: Clone,
1728    {
1729        combinations_with_replacement::combinations_with_replacement(self, k)
1730    }
1731
1732    /// Return an iterator adaptor that iterates over all k-permutations of the
1733    /// elements from an iterator.
1734    ///
1735    /// Iterator element type is `Vec<Self::Item>` with length `k`. The iterator
1736    /// produces a new `Vec` per iteration, and clones the iterator elements.
1737    ///
1738    /// If `k` is greater than the length of the input iterator, the resultant
1739    /// iterator adaptor will be empty.
1740    ///
1741    /// If you are looking for permutations with replacements,
1742    /// use `repeat_n(iter, k).multi_cartesian_product()` instead.
1743    ///
1744    /// ```
1745    /// use itertools::Itertools;
1746    ///
1747    /// let perms = (5..8).permutations(2);
1748    /// itertools::assert_equal(perms, vec![
1749    ///     vec![5, 6],
1750    ///     vec![5, 7],
1751    ///     vec![6, 5],
1752    ///     vec![6, 7],
1753    ///     vec![7, 5],
1754    ///     vec![7, 6],
1755    /// ]);
1756    /// ```
1757    ///
1758    /// Note: Permutations does not take into account the equality of the iterated values.
1759    ///
1760    /// ```
1761    /// use itertools::Itertools;
1762    ///
1763    /// let it = vec![2, 2].into_iter().permutations(2);
1764    /// itertools::assert_equal(it, vec![
1765    ///     vec![2, 2], // Note: these are the same
1766    ///     vec![2, 2], // Note: these are the same
1767    /// ]);
1768    /// ```
1769    ///
1770    /// Note: The source iterator is collected lazily, and will not be
1771    /// re-iterated if the permutations adaptor is completed and re-iterated.
1772    #[cfg(feature = "use_alloc")]
1773    fn permutations(self, k: usize) -> Permutations<Self>
1774    where
1775        Self: Sized,
1776        Self::Item: Clone,
1777    {
1778        permutations::permutations(self, k)
1779    }
1780
1781    /// Return an iterator that iterates through the powerset of the elements from an
1782    /// iterator.
1783    ///
1784    /// Iterator element type is `Vec<Self::Item>`. The iterator produces a new `Vec`
1785    /// per iteration, and clones the iterator elements.
1786    ///
1787    /// The powerset of a set contains all subsets including the empty set and the full
1788    /// input set. A powerset has length _2^n_ where _n_ is the length of the input
1789    /// set.
1790    ///
1791    /// Each `Vec` produced by this iterator represents a subset of the elements
1792    /// produced by the source iterator.
1793    ///
1794    /// ```
1795    /// use itertools::Itertools;
1796    ///
1797    /// let sets = (1..4).powerset().collect::<Vec<_>>();
1798    /// itertools::assert_equal(sets, vec![
1799    ///     vec![],
1800    ///     vec![1],
1801    ///     vec![2],
1802    ///     vec![3],
1803    ///     vec![1, 2],
1804    ///     vec![1, 3],
1805    ///     vec![2, 3],
1806    ///     vec![1, 2, 3],
1807    /// ]);
1808    /// ```
1809    #[cfg(feature = "use_alloc")]
1810    fn powerset(self) -> Powerset<Self>
1811    where
1812        Self: Sized,
1813        Self::Item: Clone,
1814    {
1815        powerset::powerset(self)
1816    }
1817
1818    /// Return an iterator adaptor that pads the sequence to a minimum length of
1819    /// `min` by filling missing elements using a closure `f`.
1820    ///
1821    /// Iterator element type is `Self::Item`.
1822    ///
1823    /// ```
1824    /// use itertools::Itertools;
1825    ///
1826    /// let it = (0..5).pad_using(10, |i| 2*i);
1827    /// itertools::assert_equal(it, vec![0, 1, 2, 3, 4, 10, 12, 14, 16, 18]);
1828    ///
1829    /// let it = (0..10).pad_using(5, |i| 2*i);
1830    /// itertools::assert_equal(it, vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9]);
1831    ///
1832    /// let it = (0..5).pad_using(10, |i| 2*i).rev();
1833    /// itertools::assert_equal(it, vec![18, 16, 14, 12, 10, 4, 3, 2, 1, 0]);
1834    /// ```
1835    fn pad_using<F>(self, min: usize, f: F) -> PadUsing<Self, F>
1836    where
1837        Self: Sized,
1838        F: FnMut(usize) -> Self::Item,
1839    {
1840        pad_tail::pad_using(self, min, f)
1841    }
1842
1843    /// Return an iterator adaptor that combines each element with a `Position` to
1844    /// ease special-case handling of the first or last elements.
1845    ///
1846    /// Iterator element type is
1847    /// [`(Position, Self::Item)`](Position)
1848    ///
1849    /// ```
1850    /// use itertools::{Itertools, Position};
1851    ///
1852    /// let it = (0..4).with_position();
1853    /// itertools::assert_equal(it,
1854    ///                         vec![(Position::First, 0),
1855    ///                              (Position::Middle, 1),
1856    ///                              (Position::Middle, 2),
1857    ///                              (Position::Last, 3)]);
1858    ///
1859    /// let it = (0..1).with_position();
1860    /// itertools::assert_equal(it, vec![(Position::Only, 0)]);
1861    /// ```
1862    fn with_position(self) -> WithPosition<Self>
1863    where
1864        Self: Sized,
1865    {
1866        with_position::with_position(self)
1867    }
1868
1869    /// Return an iterator adaptor that yields the indices of all elements
1870    /// satisfying a predicate, counted from the start of the iterator.
1871    ///
1872    /// Equivalent to `iter.enumerate().filter(|(_, v)| predicate(*v)).map(|(i, _)| i)`.
1873    ///
1874    /// ```
1875    /// use itertools::Itertools;
1876    ///
1877    /// let data = vec![1, 2, 3, 3, 4, 6, 7, 9];
1878    /// itertools::assert_equal(data.iter().positions(|v| v % 2 == 0), vec![1, 4, 5]);
1879    ///
1880    /// itertools::assert_equal(data.iter().positions(|v| v % 2 == 1).rev(), vec![7, 6, 3, 2, 0]);
1881    /// ```
1882    fn positions<P>(self, predicate: P) -> Positions<Self, P>
1883    where
1884        Self: Sized,
1885        P: FnMut(Self::Item) -> bool,
1886    {
1887        adaptors::positions(self, predicate)
1888    }
1889
1890    /// Return an iterator adaptor that applies a mutating function
1891    /// to each element before yielding it.
1892    ///
1893    /// ```
1894    /// use itertools::Itertools;
1895    ///
1896    /// let input = vec![vec![1], vec![3, 2, 1]];
1897    /// let it = input.into_iter().update(|mut v| v.push(0));
1898    /// itertools::assert_equal(it, vec![vec![1, 0], vec![3, 2, 1, 0]]);
1899    /// ```
1900    fn update<F>(self, updater: F) -> Update<Self, F>
1901    where
1902        Self: Sized,
1903        F: FnMut(&mut Self::Item),
1904    {
1905        adaptors::update(self, updater)
1906    }
1907
1908    // non-adaptor methods
1909    /// Advances the iterator and returns the next items grouped in a tuple of
1910    /// a specific size (up to 12).
1911    ///
1912    /// If there are enough elements to be grouped in a tuple, then the tuple is
1913    /// returned inside `Some`, otherwise `None` is returned.
1914    ///
1915    /// ```
1916    /// use itertools::Itertools;
1917    ///
1918    /// let mut iter = 1..5;
1919    ///
1920    /// assert_eq!(Some((1, 2)), iter.next_tuple());
1921    /// ```
1922    fn next_tuple<T>(&mut self) -> Option<T>
1923    where
1924        Self: Sized + Iterator<Item = T::Item>,
1925        T: traits::HomogeneousTuple,
1926    {
1927        T::collect_from_iter_no_buf(self)
1928    }
1929
1930    /// Collects all items from the iterator into a tuple of a specific size
1931    /// (up to 12).
1932    ///
1933    /// If the number of elements inside the iterator is **exactly** equal to
1934    /// the tuple size, then the tuple is returned inside `Some`, otherwise
1935    /// `None` is returned.
1936    ///
1937    /// ```
1938    /// use itertools::Itertools;
1939    ///
1940    /// let iter = 1..3;
1941    ///
1942    /// if let Some((x, y)) = iter.collect_tuple() {
1943    ///     assert_eq!((x, y), (1, 2))
1944    /// } else {
1945    ///     panic!("Expected two elements")
1946    /// }
1947    /// ```
1948    fn collect_tuple<T>(mut self) -> Option<T>
1949    where
1950        Self: Sized + Iterator<Item = T::Item>,
1951        T: traits::HomogeneousTuple,
1952    {
1953        match self.next_tuple() {
1954            elt @ Some(_) => match self.next() {
1955                Some(_) => None,
1956                None => elt,
1957            },
1958            _ => None,
1959        }
1960    }
1961
1962    /// Find the position and value of the first element satisfying a predicate.
1963    ///
1964    /// The iterator is not advanced past the first element found.
1965    ///
1966    /// ```
1967    /// use itertools::Itertools;
1968    ///
1969    /// let text = "Hα";
1970    /// assert_eq!(text.chars().find_position(|ch| ch.is_lowercase()), Some((1, 'α')));
1971    /// ```
1972    fn find_position<P>(&mut self, mut pred: P) -> Option<(usize, Self::Item)>
1973    where
1974        P: FnMut(&Self::Item) -> bool,
1975    {
1976        self.enumerate().find(|(_, elt)| pred(elt))
1977    }
1978    /// Find the value of the first element satisfying a predicate or return the last element, if any.
1979    ///
1980    /// The iterator is not advanced past the first element found.
1981    ///
1982    /// ```
1983    /// use itertools::Itertools;
1984    ///
1985    /// let numbers = [1, 2, 3, 4];
1986    /// assert_eq!(numbers.iter().find_or_last(|&&x| x > 5), Some(&4));
1987    /// assert_eq!(numbers.iter().find_or_last(|&&x| x > 2), Some(&3));
1988    /// assert_eq!(std::iter::empty::<i32>().find_or_last(|&x| x > 5), None);
1989    /// ```
1990    fn find_or_last<P>(mut self, mut predicate: P) -> Option<Self::Item>
1991    where
1992        Self: Sized,
1993        P: FnMut(&Self::Item) -> bool,
1994    {
1995        let mut prev = None;
1996        self.find_map(|x| {
1997            if predicate(&x) {
1998                Some(x)
1999            } else {
2000                prev = Some(x);
2001                None
2002            }
2003        })
2004        .or(prev)
2005    }
2006    /// Find the value of the first element satisfying a predicate or return the first element, if any.
2007    ///
2008    /// The iterator is not advanced past the first element found.
2009    ///
2010    /// ```
2011    /// use itertools::Itertools;
2012    ///
2013    /// let numbers = [1, 2, 3, 4];
2014    /// assert_eq!(numbers.iter().find_or_first(|&&x| x > 5), Some(&1));
2015    /// assert_eq!(numbers.iter().find_or_first(|&&x| x > 2), Some(&3));
2016    /// assert_eq!(std::iter::empty::<i32>().find_or_first(|&x| x > 5), None);
2017    /// ```
2018    fn find_or_first<P>(mut self, mut predicate: P) -> Option<Self::Item>
2019    where
2020        Self: Sized,
2021        P: FnMut(&Self::Item) -> bool,
2022    {
2023        let first = self.next()?;
2024        Some(if predicate(&first) {
2025            first
2026        } else {
2027            self.find(|x| predicate(x)).unwrap_or(first)
2028        })
2029    }
2030    /// Returns `true` if the given item is present in this iterator.
2031    ///
2032    /// This method is short-circuiting. If the given item is present in this
2033    /// iterator, this method will consume the iterator up-to-and-including
2034    /// the item. If the given item is not present in this iterator, the
2035    /// iterator will be exhausted.
2036    ///
2037    /// ```
2038    /// use itertools::Itertools;
2039    ///
2040    /// #[derive(PartialEq, Debug)]
2041    /// enum Enum { A, B, C, D, E, }
2042    ///
2043    /// let mut iter = vec![Enum::A, Enum::B, Enum::C, Enum::D].into_iter();
2044    ///
2045    /// // search `iter` for `B`
2046    /// assert_eq!(iter.contains(&Enum::B), true);
2047    /// // `B` was found, so the iterator now rests at the item after `B` (i.e, `C`).
2048    /// assert_eq!(iter.next(), Some(Enum::C));
2049    ///
2050    /// // search `iter` for `E`
2051    /// assert_eq!(iter.contains(&Enum::E), false);
2052    /// // `E` wasn't found, so `iter` is now exhausted
2053    /// assert_eq!(iter.next(), None);
2054    /// ```
2055    fn contains<Q>(&mut self, query: &Q) -> bool
2056    where
2057        Self: Sized,
2058        Self::Item: Borrow<Q>,
2059        Q: PartialEq,
2060    {
2061        self.any(|x| x.borrow() == query)
2062    }
2063
2064    /// Check whether all elements compare equal.
2065    ///
2066    /// Empty iterators are considered to have equal elements:
2067    ///
2068    /// ```
2069    /// use itertools::Itertools;
2070    ///
2071    /// let data = vec![1, 1, 1, 2, 2, 3, 3, 3, 4, 5, 5];
2072    /// assert!(!data.iter().all_equal());
2073    /// assert!(data[0..3].iter().all_equal());
2074    /// assert!(data[3..5].iter().all_equal());
2075    /// assert!(data[5..8].iter().all_equal());
2076    ///
2077    /// let data : Option<usize> = None;
2078    /// assert!(data.into_iter().all_equal());
2079    /// ```
2080    fn all_equal(&mut self) -> bool
2081    where
2082        Self: Sized,
2083        Self::Item: PartialEq,
2084    {
2085        match self.next() {
2086            None => true,
2087            Some(a) => self.all(|x| a == x),
2088        }
2089    }
2090
2091    /// If there are elements and they are all equal, return a single copy of that element.
2092    /// If there are no elements, return an Error containing None.
2093    /// If there are elements and they are not all equal, return a tuple containing the first
2094    /// two non-equal elements found.
2095    ///
2096    /// ```
2097    /// use itertools::Itertools;
2098    ///
2099    /// let data = vec![1, 1, 1, 2, 2, 3, 3, 3, 4, 5, 5];
2100    /// assert_eq!(data.iter().all_equal_value(), Err(Some((&1, &2))));
2101    /// assert_eq!(data[0..3].iter().all_equal_value(), Ok(&1));
2102    /// assert_eq!(data[3..5].iter().all_equal_value(), Ok(&2));
2103    /// assert_eq!(data[5..8].iter().all_equal_value(), Ok(&3));
2104    ///
2105    /// let data : Option<usize> = None;
2106    /// assert_eq!(data.into_iter().all_equal_value(), Err(None));
2107    /// ```
2108    #[allow(clippy::type_complexity)]
2109    fn all_equal_value(&mut self) -> Result<Self::Item, Option<(Self::Item, Self::Item)>>
2110    where
2111        Self: Sized,
2112        Self::Item: PartialEq,
2113    {
2114        let first = self.next().ok_or(None)?;
2115        let other = self.find(|x| x != &first);
2116        if let Some(other) = other {
2117            Err(Some((first, other)))
2118        } else {
2119            Ok(first)
2120        }
2121    }
2122
2123    /// Check whether all elements are unique (non equal).
2124    ///
2125    /// Empty iterators are considered to have unique elements:
2126    ///
2127    /// ```
2128    /// use itertools::Itertools;
2129    ///
2130    /// let data = vec![1, 2, 3, 4, 1, 5];
2131    /// assert!(!data.iter().all_unique());
2132    /// assert!(data[0..4].iter().all_unique());
2133    /// assert!(data[1..6].iter().all_unique());
2134    ///
2135    /// let data : Option<usize> = None;
2136    /// assert!(data.into_iter().all_unique());
2137    /// ```
2138    #[cfg(feature = "use_std")]
2139    fn all_unique(&mut self) -> bool
2140    where
2141        Self: Sized,
2142        Self::Item: Eq + Hash,
2143    {
2144        let mut used = HashSet::new();
2145        self.all(move |elt| used.insert(elt))
2146    }
2147
2148    /// Consume the first `n` elements from the iterator eagerly,
2149    /// and return the same iterator again.
2150    ///
2151    /// It works similarly to `.skip(n)` except it is eager and
2152    /// preserves the iterator type.
2153    ///
2154    /// ```
2155    /// use itertools::Itertools;
2156    ///
2157    /// let mut iter = "αβγ".chars().dropping(2);
2158    /// itertools::assert_equal(iter, "γ".chars());
2159    /// ```
2160    ///
2161    /// *Fusing notes: if the iterator is exhausted by dropping,
2162    /// the result of calling `.next()` again depends on the iterator implementation.*
2163    fn dropping(mut self, n: usize) -> Self
2164    where
2165        Self: Sized,
2166    {
2167        if n > 0 {
2168            self.nth(n - 1);
2169        }
2170        self
2171    }
2172
2173    /// Consume the last `n` elements from the iterator eagerly,
2174    /// and return the same iterator again.
2175    ///
2176    /// This is only possible on double ended iterators. `n` may be
2177    /// larger than the number of elements.
2178    ///
2179    /// Note: This method is eager, dropping the back elements immediately and
2180    /// preserves the iterator type.
2181    ///
2182    /// ```
2183    /// use itertools::Itertools;
2184    ///
2185    /// let init = vec![0, 3, 6, 9].into_iter().dropping_back(1);
2186    /// itertools::assert_equal(init, vec![0, 3, 6]);
2187    /// ```
2188    fn dropping_back(mut self, n: usize) -> Self
2189    where
2190        Self: Sized + DoubleEndedIterator,
2191    {
2192        if n > 0 {
2193            (&mut self).rev().nth(n - 1);
2194        }
2195        self
2196    }
2197
2198    /// Combine all an iterator's elements into one element by using [`Extend`].
2199    ///
2200    /// This combinator will extend the first item with each of the rest of the
2201    /// items of the iterator. If the iterator is empty, the default value of
2202    /// `I::Item` is returned.
2203    ///
2204    /// ```rust
2205    /// use itertools::Itertools;
2206    ///
2207    /// let input = vec![vec![1], vec![2, 3], vec![4, 5, 6]];
2208    /// assert_eq!(input.into_iter().concat(),
2209    ///            vec![1, 2, 3, 4, 5, 6]);
2210    /// ```
2211    fn concat(self) -> Self::Item
2212    where
2213        Self: Sized,
2214        Self::Item:
2215            Extend<<<Self as Iterator>::Item as IntoIterator>::Item> + IntoIterator + Default,
2216    {
2217        concat(self)
2218    }
2219
2220    /// `.collect_vec()` is simply a type specialization of [`Iterator::collect`],
2221    /// for convenience.
2222    #[cfg(feature = "use_alloc")]
2223    fn collect_vec(self) -> Vec<Self::Item>
2224    where
2225        Self: Sized,
2226    {
2227        self.collect()
2228    }
2229
2230    /// `.try_collect()` is more convenient way of writing
2231    /// `.collect::<Result<_, _>>()`
2232    ///
2233    /// # Example
2234    ///
2235    /// ```
2236    /// use std::{fs, io};
2237    /// use itertools::Itertools;
2238    ///
2239    /// fn process_dir_entries(entries: &[fs::DirEntry]) {
2240    ///     // ...
2241    /// }
2242    ///
2243    /// fn do_stuff() -> std::io::Result<()> {
2244    ///     let entries: Vec<_> = fs::read_dir(".")?.try_collect()?;
2245    ///     process_dir_entries(&entries);
2246    ///
2247    ///     Ok(())
2248    /// }
2249    /// ```
2250    fn try_collect<T, U, E>(self) -> Result<U, E>
2251    where
2252        Self: Sized + Iterator<Item = Result<T, E>>,
2253        Result<U, E>: FromIterator<Result<T, E>>,
2254    {
2255        self.collect()
2256    }
2257
2258    /// Assign to each reference in `self` from the `from` iterator,
2259    /// stopping at the shortest of the two iterators.
2260    ///
2261    /// The `from` iterator is queried for its next element before the `self`
2262    /// iterator, and if either is exhausted the method is done.
2263    ///
2264    /// Return the number of elements written.
2265    ///
2266    /// ```
2267    /// use itertools::Itertools;
2268    ///
2269    /// let mut xs = [0; 4];
2270    /// xs.iter_mut().set_from(1..);
2271    /// assert_eq!(xs, [1, 2, 3, 4]);
2272    /// ```
2273    #[inline]
2274    fn set_from<'a, A: 'a, J>(&mut self, from: J) -> usize
2275    where
2276        Self: Iterator<Item = &'a mut A>,
2277        J: IntoIterator<Item = A>,
2278    {
2279        from.into_iter()
2280            .zip(self)
2281            .map(|(new, old)| *old = new)
2282            .count()
2283    }
2284
2285    /// Combine all iterator elements into one `String`, separated by `sep`.
2286    ///
2287    /// Use the `Display` implementation of each element.
2288    ///
2289    /// ```
2290    /// use itertools::Itertools;
2291    ///
2292    /// assert_eq!(["a", "b", "c"].iter().join(", "), "a, b, c");
2293    /// assert_eq!([1, 2, 3].iter().join(", "), "1, 2, 3");
2294    /// ```
2295    #[cfg(feature = "use_alloc")]
2296    fn join(&mut self, sep: &str) -> String
2297    where
2298        Self::Item: std::fmt::Display,
2299    {
2300        match self.next() {
2301            None => String::new(),
2302            Some(first_elt) => {
2303                // estimate lower bound of capacity needed
2304                let (lower, _) = self.size_hint();
2305                let mut result = String::with_capacity(sep.len() * lower);
2306                write!(&mut result, "{}", first_elt).unwrap();
2307                self.for_each(|elt| {
2308                    result.push_str(sep);
2309                    write!(&mut result, "{}", elt).unwrap();
2310                });
2311                result
2312            }
2313        }
2314    }
2315
2316    /// Format all iterator elements, separated by `sep`.
2317    ///
2318    /// All elements are formatted (any formatting trait)
2319    /// with `sep` inserted between each element.
2320    ///
2321    /// **Panics** if the formatter helper is formatted more than once.
2322    ///
2323    /// ```
2324    /// use itertools::Itertools;
2325    ///
2326    /// let data = [1.1, 2.71828, -3.];
2327    /// assert_eq!(
2328    ///     format!("{:.2}", data.iter().format(", ")),
2329    ///            "1.10, 2.72, -3.00");
2330    /// ```
2331    fn format(self, sep: &str) -> Format<Self>
2332    where
2333        Self: Sized,
2334    {
2335        format::new_format_default(self, sep)
2336    }
2337
2338    /// Format all iterator elements, separated by `sep`.
2339    ///
2340    /// This is a customizable version of [`.format()`](Itertools::format).
2341    ///
2342    /// The supplied closure `format` is called once per iterator element,
2343    /// with two arguments: the element and a callback that takes a
2344    /// `&Display` value, i.e. any reference to type that implements `Display`.
2345    ///
2346    /// Using `&format_args!(...)` is the most versatile way to apply custom
2347    /// element formatting. The callback can be called multiple times if needed.
2348    ///
2349    /// **Panics** if the formatter helper is formatted more than once.
2350    ///
2351    /// ```
2352    /// use itertools::Itertools;
2353    ///
2354    /// let data = [1.1, 2.71828, -3.];
2355    /// let data_formatter = data.iter().format_with(", ", |elt, f| f(&format_args!("{:.2}", elt)));
2356    /// assert_eq!(format!("{}", data_formatter),
2357    ///            "1.10, 2.72, -3.00");
2358    ///
2359    /// // .format_with() is recursively composable
2360    /// let matrix = [[1., 2., 3.],
2361    ///               [4., 5., 6.]];
2362    /// let matrix_formatter = matrix.iter().format_with("\n", |row, f| {
2363    ///                                 f(&row.iter().format_with(", ", |elt, g| g(&elt)))
2364    ///                              });
2365    /// assert_eq!(format!("{}", matrix_formatter),
2366    ///            "1, 2, 3\n4, 5, 6");
2367    ///
2368    ///
2369    /// ```
2370    fn format_with<F>(self, sep: &str, format: F) -> FormatWith<Self, F>
2371    where
2372        Self: Sized,
2373        F: FnMut(Self::Item, &mut dyn FnMut(&dyn fmt::Display) -> fmt::Result) -> fmt::Result,
2374    {
2375        format::new_format(self, sep, format)
2376    }
2377
2378    /// Fold `Result` values from an iterator.
2379    ///
2380    /// Only `Ok` values are folded. If no error is encountered, the folded
2381    /// value is returned inside `Ok`. Otherwise, the operation terminates
2382    /// and returns the first `Err` value it encounters. No iterator elements are
2383    /// consumed after the first error.
2384    ///
2385    /// The first accumulator value is the `start` parameter.
2386    /// Each iteration passes the accumulator value and the next value inside `Ok`
2387    /// to the fold function `f` and its return value becomes the new accumulator value.
2388    ///
2389    /// For example the sequence *Ok(1), Ok(2), Ok(3)* will result in a
2390    /// computation like this:
2391    ///
2392    /// ```no_run
2393    /// # let start = 0;
2394    /// # let f = |x, y| x + y;
2395    /// let mut accum = start;
2396    /// accum = f(accum, 1);
2397    /// accum = f(accum, 2);
2398    /// accum = f(accum, 3);
2399    /// ```
2400    ///
2401    /// With a `start` value of 0 and an addition as folding function,
2402    /// this effectively results in *((0 + 1) + 2) + 3*
2403    ///
2404    /// ```
2405    /// use std::ops::Add;
2406    /// use itertools::Itertools;
2407    ///
2408    /// let values = [1, 2, -2, -1, 2, 1];
2409    /// assert_eq!(
2410    ///     values.iter()
2411    ///           .map(Ok::<_, ()>)
2412    ///           .fold_ok(0, Add::add),
2413    ///     Ok(3)
2414    /// );
2415    /// assert!(
2416    ///     values.iter()
2417    ///           .map(|&x| if x >= 0 { Ok(x) } else { Err("Negative number") })
2418    ///           .fold_ok(0, Add::add)
2419    ///           .is_err()
2420    /// );
2421    /// ```
2422    fn fold_ok<A, E, B, F>(&mut self, mut start: B, mut f: F) -> Result<B, E>
2423    where
2424        Self: Iterator<Item = Result<A, E>>,
2425        F: FnMut(B, A) -> B,
2426    {
2427        for elt in self {
2428            match elt {
2429                Ok(v) => start = f(start, v),
2430                Err(u) => return Err(u),
2431            }
2432        }
2433        Ok(start)
2434    }
2435
2436    /// Fold `Option` values from an iterator.
2437    ///
2438    /// Only `Some` values are folded. If no `None` is encountered, the folded
2439    /// value is returned inside `Some`. Otherwise, the operation terminates
2440    /// and returns `None`. No iterator elements are consumed after the `None`.
2441    ///
2442    /// This is the `Option` equivalent to [`fold_ok`](Itertools::fold_ok).
2443    ///
2444    /// ```
2445    /// use std::ops::Add;
2446    /// use itertools::Itertools;
2447    ///
2448    /// let mut values = vec![Some(1), Some(2), Some(-2)].into_iter();
2449    /// assert_eq!(values.fold_options(5, Add::add), Some(5 + 1 + 2 - 2));
2450    ///
2451    /// let mut more_values = vec![Some(2), None, Some(0)].into_iter();
2452    /// assert!(more_values.fold_options(0, Add::add).is_none());
2453    /// assert_eq!(more_values.next().unwrap(), Some(0));
2454    /// ```
2455    fn fold_options<A, B, F>(&mut self, mut start: B, mut f: F) -> Option<B>
2456    where
2457        Self: Iterator<Item = Option<A>>,
2458        F: FnMut(B, A) -> B,
2459    {
2460        for elt in self {
2461            match elt {
2462                Some(v) => start = f(start, v),
2463                None => return None,
2464            }
2465        }
2466        Some(start)
2467    }
2468
2469    /// Accumulator of the elements in the iterator.
2470    ///
2471    /// Like `.fold()`, without a base case. If the iterator is
2472    /// empty, return `None`. With just one element, return it.
2473    /// Otherwise elements are accumulated in sequence using the closure `f`.
2474    ///
2475    /// ```
2476    /// use itertools::Itertools;
2477    ///
2478    /// assert_eq!((0..10).fold1(|x, y| x + y).unwrap_or(0), 45);
2479    /// assert_eq!((0..0).fold1(|x, y| x * y), None);
2480    /// ```
2481    #[deprecated(
2482        note = "Use [`Iterator::reduce`](https://doc.rust-lang.org/std/iter/trait.Iterator.html#method.reduce) instead",
2483        since = "0.10.2"
2484    )]
2485    fn fold1<F>(mut self, f: F) -> Option<Self::Item>
2486    where
2487        F: FnMut(Self::Item, Self::Item) -> Self::Item,
2488        Self: Sized,
2489    {
2490        self.next().map(move |x| self.fold(x, f))
2491    }
2492
2493    /// Accumulate the elements in the iterator in a tree-like manner.
2494    ///
2495    /// You can think of it as, while there's more than one item, repeatedly
2496    /// combining adjacent items.  It does so in bottom-up-merge-sort order,
2497    /// however, so that it needs only logarithmic stack space.
2498    ///
2499    /// This produces a call tree like the following (where the calls under
2500    /// an item are done after reading that item):
2501    ///
2502    /// ```text
2503    /// 1 2 3 4 5 6 7
2504    /// │ │ │ │ │ │ │
2505    /// └─f └─f └─f │
2506    ///   │   │   │ │
2507    ///   └───f   └─f
2508    ///       │     │
2509    ///       └─────f
2510    /// ```
2511    ///
2512    /// Which, for non-associative functions, will typically produce a different
2513    /// result than the linear call tree used by [`Iterator::reduce`]:
2514    ///
2515    /// ```text
2516    /// 1 2 3 4 5 6 7
2517    /// │ │ │ │ │ │ │
2518    /// └─f─f─f─f─f─f
2519    /// ```
2520    ///
2521    /// If `f` is associative you should also decide carefully:
2522    ///
2523    /// - if `f` is a trivial operation like `u32::wrapping_add`, prefer the normal
2524    /// [`Iterator::reduce`] instead since it will most likely result in the generation of simpler
2525    /// code because the compiler is able to optimize it
2526    /// - otherwise if `f` is non-trivial like `format!`, you should use `tree_reduce` since it
2527    /// reduces the number of operations from `O(n)` to `O(ln(n))`
2528    ///
2529    /// Here "non-trivial" means:
2530    ///
2531    /// - any allocating operation
2532    /// - any function that is a composition of many operations
2533    ///
2534    /// ```
2535    /// use itertools::Itertools;
2536    ///
2537    /// // The same tree as above
2538    /// let num_strings = (1..8).map(|x| x.to_string());
2539    /// assert_eq!(num_strings.tree_reduce(|x, y| format!("f({}, {})", x, y)),
2540    ///     Some(String::from("f(f(f(1, 2), f(3, 4)), f(f(5, 6), 7))")));
2541    ///
2542    /// // Like fold1, an empty iterator produces None
2543    /// assert_eq!((0..0).tree_reduce(|x, y| x * y), None);
2544    ///
2545    /// // tree_reduce matches fold1 for associative operations...
2546    /// assert_eq!((0..10).tree_reduce(|x, y| x + y),
2547    ///     (0..10).fold1(|x, y| x + y));
2548    /// // ...but not for non-associative ones
2549    /// assert_ne!((0..10).tree_reduce(|x, y| x - y),
2550    ///     (0..10).fold1(|x, y| x - y));
2551    /// ```
2552    fn tree_reduce<F>(mut self, mut f: F) -> Option<Self::Item>
2553    where
2554        F: FnMut(Self::Item, Self::Item) -> Self::Item,
2555        Self: Sized,
2556    {
2557        type State<T> = Result<T, Option<T>>;
2558
2559        fn inner0<T, II, FF>(it: &mut II, f: &mut FF) -> State<T>
2560        where
2561            II: Iterator<Item = T>,
2562            FF: FnMut(T, T) -> T,
2563        {
2564            // This function could be replaced with `it.next().ok_or(None)`,
2565            // but half the useful tree_reduce work is combining adjacent items,
2566            // so put that in a form that LLVM is more likely to optimize well.
2567
2568            let a = if let Some(v) = it.next() {
2569                v
2570            } else {
2571                return Err(None);
2572            };
2573            let b = if let Some(v) = it.next() {
2574                v
2575            } else {
2576                return Err(Some(a));
2577            };
2578            Ok(f(a, b))
2579        }
2580
2581        fn inner<T, II, FF>(stop: usize, it: &mut II, f: &mut FF) -> State<T>
2582        where
2583            II: Iterator<Item = T>,
2584            FF: FnMut(T, T) -> T,
2585        {
2586            let mut x = inner0(it, f)?;
2587            for height in 0..stop {
2588                // Try to get another tree the same size with which to combine it,
2589                // creating a new tree that's twice as big for next time around.
2590                let next = if height == 0 {
2591                    inner0(it, f)
2592                } else {
2593                    inner(height, it, f)
2594                };
2595                match next {
2596                    Ok(y) => x = f(x, y),
2597
2598                    // If we ran out of items, combine whatever we did manage
2599                    // to get.  It's better combined with the current value
2600                    // than something in a parent frame, because the tree in
2601                    // the parent is always as least as big as this one.
2602                    Err(None) => return Err(Some(x)),
2603                    Err(Some(y)) => return Err(Some(f(x, y))),
2604                }
2605            }
2606            Ok(x)
2607        }
2608
2609        match inner(usize::MAX, &mut self, &mut f) {
2610            Err(x) => x,
2611            _ => unreachable!(),
2612        }
2613    }
2614
2615    /// See [`.tree_reduce()`](Itertools::tree_reduce).
2616    #[deprecated(note = "Use .tree_reduce() instead", since = "0.13.0")]
2617    fn tree_fold1<F>(self, f: F) -> Option<Self::Item>
2618    where
2619        F: FnMut(Self::Item, Self::Item) -> Self::Item,
2620        Self: Sized,
2621    {
2622        self.tree_reduce(f)
2623    }
2624
2625    /// An iterator method that applies a function, producing a single, final value.
2626    ///
2627    /// `fold_while()` is basically equivalent to [`Iterator::fold`] but with additional support for
2628    /// early exit via short-circuiting.
2629    ///
2630    /// ```
2631    /// use itertools::Itertools;
2632    /// use itertools::FoldWhile::{Continue, Done};
2633    ///
2634    /// let numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
2635    ///
2636    /// let mut result = 0;
2637    ///
2638    /// // for loop:
2639    /// for i in &numbers {
2640    ///     if *i > 5 {
2641    ///         break;
2642    ///     }
2643    ///     result = result + i;
2644    /// }
2645    ///
2646    /// // fold:
2647    /// let result2 = numbers.iter().fold(0, |acc, x| {
2648    ///     if *x > 5 { acc } else { acc + x }
2649    /// });
2650    ///
2651    /// // fold_while:
2652    /// let result3 = numbers.iter().fold_while(0, |acc, x| {
2653    ///     if *x > 5 { Done(acc) } else { Continue(acc + x) }
2654    /// }).into_inner();
2655    ///
2656    /// // they're the same
2657    /// assert_eq!(result, result2);
2658    /// assert_eq!(result2, result3);
2659    /// ```
2660    ///
2661    /// The big difference between the computations of `result2` and `result3` is that while
2662    /// `fold()` called the provided closure for every item of the callee iterator,
2663    /// `fold_while()` actually stopped iterating as soon as it encountered `Fold::Done(_)`.
2664    fn fold_while<B, F>(&mut self, init: B, mut f: F) -> FoldWhile<B>
2665    where
2666        Self: Sized,
2667        F: FnMut(B, Self::Item) -> FoldWhile<B>,
2668    {
2669        use Result::{Err as Break, Ok as Continue};
2670
2671        let result = self.try_fold(
2672            init,
2673            #[inline(always)]
2674            |acc, v| match f(acc, v) {
2675                FoldWhile::Continue(acc) => Continue(acc),
2676                FoldWhile::Done(acc) => Break(acc),
2677            },
2678        );
2679
2680        match result {
2681            Continue(acc) => FoldWhile::Continue(acc),
2682            Break(acc) => FoldWhile::Done(acc),
2683        }
2684    }
2685
2686    /// Iterate over the entire iterator and add all the elements.
2687    ///
2688    /// An empty iterator returns `None`, otherwise `Some(sum)`.
2689    ///
2690    /// # Panics
2691    ///
2692    /// When calling `sum1()` and a primitive integer type is being returned, this
2693    /// method will panic if the computation overflows and debug assertions are
2694    /// enabled.
2695    ///
2696    /// # Examples
2697    ///
2698    /// ```
2699    /// use itertools::Itertools;
2700    ///
2701    /// let empty_sum = (1..1).sum1::<i32>();
2702    /// assert_eq!(empty_sum, None);
2703    ///
2704    /// let nonempty_sum = (1..11).sum1::<i32>();
2705    /// assert_eq!(nonempty_sum, Some(55));
2706    /// ```
2707    fn sum1<S>(mut self) -> Option<S>
2708    where
2709        Self: Sized,
2710        S: std::iter::Sum<Self::Item>,
2711    {
2712        self.next().map(|first| once(first).chain(self).sum())
2713    }
2714
2715    /// Iterate over the entire iterator and multiply all the elements.
2716    ///
2717    /// An empty iterator returns `None`, otherwise `Some(product)`.
2718    ///
2719    /// # Panics
2720    ///
2721    /// When calling `product1()` and a primitive integer type is being returned,
2722    /// method will panic if the computation overflows and debug assertions are
2723    /// enabled.
2724    ///
2725    /// # Examples
2726    /// ```
2727    /// use itertools::Itertools;
2728    ///
2729    /// let empty_product = (1..1).product1::<i32>();
2730    /// assert_eq!(empty_product, None);
2731    ///
2732    /// let nonempty_product = (1..11).product1::<i32>();
2733    /// assert_eq!(nonempty_product, Some(3628800));
2734    /// ```
2735    fn product1<P>(mut self) -> Option<P>
2736    where
2737        Self: Sized,
2738        P: std::iter::Product<Self::Item>,
2739    {
2740        self.next().map(|first| once(first).chain(self).product())
2741    }
2742
2743    /// Sort all iterator elements into a new iterator in ascending order.
2744    ///
2745    /// **Note:** This consumes the entire iterator, uses the
2746    /// [`slice::sort_unstable`] method and returns the result as a new
2747    /// iterator that owns its elements.
2748    ///
2749    /// This sort is unstable (i.e., may reorder equal elements).
2750    ///
2751    /// The sorted iterator, if directly collected to a `Vec`, is converted
2752    /// without any extra copying or allocation cost.
2753    ///
2754    /// ```
2755    /// use itertools::Itertools;
2756    ///
2757    /// // sort the letters of the text in ascending order
2758    /// let text = "bdacfe";
2759    /// itertools::assert_equal(text.chars().sorted_unstable(),
2760    ///                         "abcdef".chars());
2761    /// ```
2762    #[cfg(feature = "use_alloc")]
2763    fn sorted_unstable(self) -> VecIntoIter<Self::Item>
2764    where
2765        Self: Sized,
2766        Self::Item: Ord,
2767    {
2768        // Use .sort_unstable() directly since it is not quite identical with
2769        // .sort_by(Ord::cmp)
2770        let mut v = Vec::from_iter(self);
2771        v.sort_unstable();
2772        v.into_iter()
2773    }
2774
2775    /// Sort all iterator elements into a new iterator in ascending order.
2776    ///
2777    /// **Note:** This consumes the entire iterator, uses the
2778    /// [`slice::sort_unstable_by`] method and returns the result as a new
2779    /// iterator that owns its elements.
2780    ///
2781    /// This sort is unstable (i.e., may reorder equal elements).
2782    ///
2783    /// The sorted iterator, if directly collected to a `Vec`, is converted
2784    /// without any extra copying or allocation cost.
2785    ///
2786    /// ```
2787    /// use itertools::Itertools;
2788    ///
2789    /// // sort people in descending order by age
2790    /// let people = vec![("Jane", 20), ("John", 18), ("Jill", 30), ("Jack", 27)];
2791    ///
2792    /// let oldest_people_first = people
2793    ///     .into_iter()
2794    ///     .sorted_unstable_by(|a, b| Ord::cmp(&b.1, &a.1))
2795    ///     .map(|(person, _age)| person);
2796    ///
2797    /// itertools::assert_equal(oldest_people_first,
2798    ///                         vec!["Jill", "Jack", "Jane", "John"]);
2799    /// ```
2800    #[cfg(feature = "use_alloc")]
2801    fn sorted_unstable_by<F>(self, cmp: F) -> VecIntoIter<Self::Item>
2802    where
2803        Self: Sized,
2804        F: FnMut(&Self::Item, &Self::Item) -> Ordering,
2805    {
2806        let mut v = Vec::from_iter(self);
2807        v.sort_unstable_by(cmp);
2808        v.into_iter()
2809    }
2810
2811    /// Sort all iterator elements into a new iterator in ascending order.
2812    ///
2813    /// **Note:** This consumes the entire iterator, uses the
2814    /// [`slice::sort_unstable_by_key`] method and returns the result as a new
2815    /// iterator that owns its elements.
2816    ///
2817    /// This sort is unstable (i.e., may reorder equal elements).
2818    ///
2819    /// The sorted iterator, if directly collected to a `Vec`, is converted
2820    /// without any extra copying or allocation cost.
2821    ///
2822    /// ```
2823    /// use itertools::Itertools;
2824    ///
2825    /// // sort people in descending order by age
2826    /// let people = vec![("Jane", 20), ("John", 18), ("Jill", 30), ("Jack", 27)];
2827    ///
2828    /// let oldest_people_first = people
2829    ///     .into_iter()
2830    ///     .sorted_unstable_by_key(|x| -x.1)
2831    ///     .map(|(person, _age)| person);
2832    ///
2833    /// itertools::assert_equal(oldest_people_first,
2834    ///                         vec!["Jill", "Jack", "Jane", "John"]);
2835    /// ```
2836    #[cfg(feature = "use_alloc")]
2837    fn sorted_unstable_by_key<K, F>(self, f: F) -> VecIntoIter<Self::Item>
2838    where
2839        Self: Sized,
2840        K: Ord,
2841        F: FnMut(&Self::Item) -> K,
2842    {
2843        let mut v = Vec::from_iter(self);
2844        v.sort_unstable_by_key(f);
2845        v.into_iter()
2846    }
2847
2848    /// Sort all iterator elements into a new iterator in ascending order.
2849    ///
2850    /// **Note:** This consumes the entire iterator, uses the
2851    /// [`slice::sort`] method and returns the result as a new
2852    /// iterator that owns its elements.
2853    ///
2854    /// This sort is stable (i.e., does not reorder equal elements).
2855    ///
2856    /// The sorted iterator, if directly collected to a `Vec`, is converted
2857    /// without any extra copying or allocation cost.
2858    ///
2859    /// ```
2860    /// use itertools::Itertools;
2861    ///
2862    /// // sort the letters of the text in ascending order
2863    /// let text = "bdacfe";
2864    /// itertools::assert_equal(text.chars().sorted(),
2865    ///                         "abcdef".chars());
2866    /// ```
2867    #[cfg(feature = "use_alloc")]
2868    fn sorted(self) -> VecIntoIter<Self::Item>
2869    where
2870        Self: Sized,
2871        Self::Item: Ord,
2872    {
2873        // Use .sort() directly since it is not quite identical with
2874        // .sort_by(Ord::cmp)
2875        let mut v = Vec::from_iter(self);
2876        v.sort();
2877        v.into_iter()
2878    }
2879
2880    /// Sort all iterator elements into a new iterator in ascending order.
2881    ///
2882    /// **Note:** This consumes the entire iterator, uses the
2883    /// [`slice::sort_by`] method and returns the result as a new
2884    /// iterator that owns its elements.
2885    ///
2886    /// This sort is stable (i.e., does not reorder equal elements).
2887    ///
2888    /// The sorted iterator, if directly collected to a `Vec`, is converted
2889    /// without any extra copying or allocation cost.
2890    ///
2891    /// ```
2892    /// use itertools::Itertools;
2893    ///
2894    /// // sort people in descending order by age
2895    /// let people = vec![("Jane", 20), ("John", 18), ("Jill", 30), ("Jack", 30)];
2896    ///
2897    /// let oldest_people_first = people
2898    ///     .into_iter()
2899    ///     .sorted_by(|a, b| Ord::cmp(&b.1, &a.1))
2900    ///     .map(|(person, _age)| person);
2901    ///
2902    /// itertools::assert_equal(oldest_people_first,
2903    ///                         vec!["Jill", "Jack", "Jane", "John"]);
2904    /// ```
2905    #[cfg(feature = "use_alloc")]
2906    fn sorted_by<F>(self, cmp: F) -> VecIntoIter<Self::Item>
2907    where
2908        Self: Sized,
2909        F: FnMut(&Self::Item, &Self::Item) -> Ordering,
2910    {
2911        let mut v = Vec::from_iter(self);
2912        v.sort_by(cmp);
2913        v.into_iter()
2914    }
2915
2916    /// Sort all iterator elements into a new iterator in ascending order.
2917    ///
2918    /// **Note:** This consumes the entire iterator, uses the
2919    /// [`slice::sort_by_key`] method and returns the result as a new
2920    /// iterator that owns its elements.
2921    ///
2922    /// This sort is stable (i.e., does not reorder equal elements).
2923    ///
2924    /// The sorted iterator, if directly collected to a `Vec`, is converted
2925    /// without any extra copying or allocation cost.
2926    ///
2927    /// ```
2928    /// use itertools::Itertools;
2929    ///
2930    /// // sort people in descending order by age
2931    /// let people = vec![("Jane", 20), ("John", 18), ("Jill", 30), ("Jack", 30)];
2932    ///
2933    /// let oldest_people_first = people
2934    ///     .into_iter()
2935    ///     .sorted_by_key(|x| -x.1)
2936    ///     .map(|(person, _age)| person);
2937    ///
2938    /// itertools::assert_equal(oldest_people_first,
2939    ///                         vec!["Jill", "Jack", "Jane", "John"]);
2940    /// ```
2941    #[cfg(feature = "use_alloc")]
2942    fn sorted_by_key<K, F>(self, f: F) -> VecIntoIter<Self::Item>
2943    where
2944        Self: Sized,
2945        K: Ord,
2946        F: FnMut(&Self::Item) -> K,
2947    {
2948        let mut v = Vec::from_iter(self);
2949        v.sort_by_key(f);
2950        v.into_iter()
2951    }
2952
2953    /// Sort all iterator elements into a new iterator in ascending order. The key function is
2954    /// called exactly once per key.
2955    ///
2956    /// **Note:** This consumes the entire iterator, uses the
2957    /// [`slice::sort_by_cached_key`] method and returns the result as a new
2958    /// iterator that owns its elements.
2959    ///
2960    /// This sort is stable (i.e., does not reorder equal elements).
2961    ///
2962    /// The sorted iterator, if directly collected to a `Vec`, is converted
2963    /// without any extra copying or allocation cost.
2964    ///
2965    /// ```
2966    /// use itertools::Itertools;
2967    ///
2968    /// // sort people in descending order by age
2969    /// let people = vec![("Jane", 20), ("John", 18), ("Jill", 30), ("Jack", 30)];
2970    ///
2971    /// let oldest_people_first = people
2972    ///     .into_iter()
2973    ///     .sorted_by_cached_key(|x| -x.1)
2974    ///     .map(|(person, _age)| person);
2975    ///
2976    /// itertools::assert_equal(oldest_people_first,
2977    ///                         vec!["Jill", "Jack", "Jane", "John"]);
2978    /// ```
2979    #[cfg(feature = "use_alloc")]
2980    fn sorted_by_cached_key<K, F>(self, f: F) -> VecIntoIter<Self::Item>
2981    where
2982        Self: Sized,
2983        K: Ord,
2984        F: FnMut(&Self::Item) -> K,
2985    {
2986        let mut v = Vec::from_iter(self);
2987        v.sort_by_cached_key(f);
2988        v.into_iter()
2989    }
2990
2991    /// Sort the k smallest elements into a new iterator, in ascending order.
2992    ///
2993    /// **Note:** This consumes the entire iterator, and returns the result
2994    /// as a new iterator that owns its elements.  If the input contains
2995    /// less than k elements, the result is equivalent to `self.sorted()`.
2996    ///
2997    /// This is guaranteed to use `k * sizeof(Self::Item) + O(1)` memory
2998    /// and `O(n log k)` time, with `n` the number of elements in the input.
2999    ///
3000    /// The sorted iterator, if directly collected to a `Vec`, is converted
3001    /// without any extra copying or allocation cost.
3002    ///
3003    /// **Note:** This is functionally-equivalent to `self.sorted().take(k)`
3004    /// but much more efficient.
3005    ///
3006    /// ```
3007    /// use itertools::Itertools;
3008    ///
3009    /// // A random permutation of 0..15
3010    /// let numbers = vec![6, 9, 1, 14, 0, 4, 8, 7, 11, 2, 10, 3, 13, 12, 5];
3011    ///
3012    /// let five_smallest = numbers
3013    ///     .into_iter()
3014    ///     .k_smallest(5);
3015    ///
3016    /// itertools::assert_equal(five_smallest, 0..5);
3017    /// ```
3018    #[cfg(feature = "use_alloc")]
3019    fn k_smallest(self, k: usize) -> VecIntoIter<Self::Item>
3020    where
3021        Self: Sized,
3022        Self::Item: Ord,
3023    {
3024        // The stdlib heap has optimised handling of "holes", which is not included in our heap implementation in k_smallest_general.
3025        // While the difference is unlikely to have practical impact unless `Self::Item` is very large, this method uses the stdlib structure
3026        // to maintain performance compared to previous versions of the crate.
3027        use alloc::collections::BinaryHeap;
3028
3029        if k == 0 {
3030            self.last();
3031            return Vec::new().into_iter();
3032        }
3033        if k == 1 {
3034            return self.min().into_iter().collect_vec().into_iter();
3035        }
3036
3037        let mut iter = self.fuse();
3038        let mut heap: BinaryHeap<_> = iter.by_ref().take(k).collect();
3039
3040        iter.for_each(|i| {
3041            debug_assert_eq!(heap.len(), k);
3042            // Equivalent to heap.push(min(i, heap.pop())) but more efficient.
3043            // This should be done with a single `.peek_mut().unwrap()` but
3044            //  `PeekMut` sifts-down unconditionally on Rust 1.46.0 and prior.
3045            if *heap.peek().unwrap() > i {
3046                *heap.peek_mut().unwrap() = i;
3047            }
3048        });
3049
3050        heap.into_sorted_vec().into_iter()
3051    }
3052
3053    /// Sort the k smallest elements into a new iterator using the provided comparison.
3054    ///
3055    /// The sorted iterator, if directly collected to a `Vec`, is converted
3056    /// without any extra copying or allocation cost.
3057    ///
3058    /// This corresponds to `self.sorted_by(cmp).take(k)` in the same way that
3059    /// [`k_smallest`](Itertools::k_smallest) corresponds to `self.sorted().take(k)`,
3060    /// in both semantics and complexity.
3061    ///
3062    /// Particularly, a custom heap implementation ensures the comparison is not cloned.
3063    ///
3064    /// ```
3065    /// use itertools::Itertools;
3066    ///
3067    /// // A random permutation of 0..15
3068    /// let numbers = vec![6, 9, 1, 14, 0, 4, 8, 7, 11, 2, 10, 3, 13, 12, 5];
3069    ///
3070    /// let five_smallest = numbers
3071    ///     .into_iter()
3072    ///     .k_smallest_by(5, |a, b| (a % 7).cmp(&(b % 7)).then(a.cmp(b)));
3073    ///
3074    /// itertools::assert_equal(five_smallest, vec![0, 7, 14, 1, 8]);
3075    /// ```
3076    #[cfg(feature = "use_alloc")]
3077    fn k_smallest_by<F>(self, k: usize, cmp: F) -> VecIntoIter<Self::Item>
3078    where
3079        Self: Sized,
3080        F: FnMut(&Self::Item, &Self::Item) -> Ordering,
3081    {
3082        k_smallest::k_smallest_general(self, k, cmp).into_iter()
3083    }
3084
3085    /// Return the elements producing the k smallest outputs of the provided function.
3086    ///
3087    /// The sorted iterator, if directly collected to a `Vec`, is converted
3088    /// without any extra copying or allocation cost.
3089    ///
3090    /// This corresponds to `self.sorted_by_key(key).take(k)` in the same way that
3091    /// [`k_smallest`](Itertools::k_smallest) corresponds to `self.sorted().take(k)`,
3092    /// in both semantics and complexity.
3093    ///
3094    /// Particularly, a custom heap implementation ensures the comparison is not cloned.
3095    ///
3096    /// ```
3097    /// use itertools::Itertools;
3098    ///
3099    /// // A random permutation of 0..15
3100    /// let numbers = vec![6, 9, 1, 14, 0, 4, 8, 7, 11, 2, 10, 3, 13, 12, 5];
3101    ///
3102    /// let five_smallest = numbers
3103    ///     .into_iter()
3104    ///     .k_smallest_by_key(5, |n| (n % 7, *n));
3105    ///
3106    /// itertools::assert_equal(five_smallest, vec![0, 7, 14, 1, 8]);
3107    /// ```
3108    #[cfg(feature = "use_alloc")]
3109    fn k_smallest_by_key<F, K>(self, k: usize, key: F) -> VecIntoIter<Self::Item>
3110    where
3111        Self: Sized,
3112        F: FnMut(&Self::Item) -> K,
3113        K: Ord,
3114    {
3115        self.k_smallest_by(k, k_smallest::key_to_cmp(key))
3116    }
3117
3118    /// Sort the k largest elements into a new iterator, in descending order.
3119    ///
3120    /// The sorted iterator, if directly collected to a `Vec`, is converted
3121    /// without any extra copying or allocation cost.
3122    ///
3123    /// It is semantically equivalent to [`k_smallest`](Itertools::k_smallest)
3124    /// with a reversed `Ord`.
3125    /// However, this is implemented with a custom binary heap which does not
3126    /// have the same performance characteristics for very large `Self::Item`.
3127    ///
3128    /// ```
3129    /// use itertools::Itertools;
3130    ///
3131    /// // A random permutation of 0..15
3132    /// let numbers = vec![6, 9, 1, 14, 0, 4, 8, 7, 11, 2, 10, 3, 13, 12, 5];
3133    ///
3134    /// let five_largest = numbers
3135    ///     .into_iter()
3136    ///     .k_largest(5);
3137    ///
3138    /// itertools::assert_equal(five_largest, vec![14, 13, 12, 11, 10]);
3139    /// ```
3140    #[cfg(feature = "use_alloc")]
3141    fn k_largest(self, k: usize) -> VecIntoIter<Self::Item>
3142    where
3143        Self: Sized,
3144        Self::Item: Ord,
3145    {
3146        self.k_largest_by(k, Self::Item::cmp)
3147    }
3148
3149    /// Sort the k largest elements into a new iterator using the provided comparison.
3150    ///
3151    /// The sorted iterator, if directly collected to a `Vec`, is converted
3152    /// without any extra copying or allocation cost.
3153    ///
3154    /// Functionally equivalent to [`k_smallest_by`](Itertools::k_smallest_by)
3155    /// with a reversed `Ord`.
3156    ///
3157    /// ```
3158    /// use itertools::Itertools;
3159    ///
3160    /// // A random permutation of 0..15
3161    /// let numbers = vec![6, 9, 1, 14, 0, 4, 8, 7, 11, 2, 10, 3, 13, 12, 5];
3162    ///
3163    /// let five_largest = numbers
3164    ///     .into_iter()
3165    ///     .k_largest_by(5, |a, b| (a % 7).cmp(&(b % 7)).then(a.cmp(b)));
3166    ///
3167    /// itertools::assert_equal(five_largest, vec![13, 6, 12, 5, 11]);
3168    /// ```
3169    #[cfg(feature = "use_alloc")]
3170    fn k_largest_by<F>(self, k: usize, mut cmp: F) -> VecIntoIter<Self::Item>
3171    where
3172        Self: Sized,
3173        F: FnMut(&Self::Item, &Self::Item) -> Ordering,
3174    {
3175        self.k_smallest_by(k, move |a, b| cmp(b, a))
3176    }
3177
3178    /// Return the elements producing the k largest outputs of the provided function.
3179    ///
3180    /// The sorted iterator, if directly collected to a `Vec`, is converted
3181    /// without any extra copying or allocation cost.
3182    ///
3183    /// Functionally equivalent to [`k_smallest_by_key`](Itertools::k_smallest_by_key)
3184    /// with a reversed `Ord`.
3185    ///
3186    /// ```
3187    /// use itertools::Itertools;
3188    ///
3189    /// // A random permutation of 0..15
3190    /// let numbers = vec![6, 9, 1, 14, 0, 4, 8, 7, 11, 2, 10, 3, 13, 12, 5];
3191    ///
3192    /// let five_largest = numbers
3193    ///     .into_iter()
3194    ///     .k_largest_by_key(5, |n| (n % 7, *n));
3195    ///
3196    /// itertools::assert_equal(five_largest, vec![13, 6, 12, 5, 11]);
3197    /// ```
3198    #[cfg(feature = "use_alloc")]
3199    fn k_largest_by_key<F, K>(self, k: usize, key: F) -> VecIntoIter<Self::Item>
3200    where
3201        Self: Sized,
3202        F: FnMut(&Self::Item) -> K,
3203        K: Ord,
3204    {
3205        self.k_largest_by(k, k_smallest::key_to_cmp(key))
3206    }
3207
3208    /// Consumes the iterator and return an iterator of the last `n` elements.
3209    ///
3210    /// The iterator, if directly collected to a `VecDeque`, is converted
3211    /// without any extra copying or allocation cost.
3212    /// If directly collected to a `Vec`, it may need some data movement
3213    /// but no re-allocation.
3214    ///
3215    /// ```
3216    /// use itertools::{assert_equal, Itertools};
3217    ///
3218    /// let v = vec![5, 9, 8, 4, 2, 12, 0];
3219    /// assert_equal(v.iter().tail(3), &[2, 12, 0]);
3220    /// assert_equal(v.iter().tail(10), &v);
3221    ///
3222    /// assert_equal(v.iter().tail(1), v.iter().last());
3223    ///
3224    /// assert_equal((0..100).tail(10), 90..100);
3225    ///
3226    /// assert_equal((0..100).filter(|x| x % 3 == 0).tail(10), (72..100).step_by(3));
3227    /// ```
3228    ///
3229    /// For double ended iterators without side-effects, you might prefer
3230    /// `.rev().take(n).rev()` to have a similar result (lazy and non-allocating)
3231    /// without consuming the entire iterator.
3232    #[cfg(feature = "use_alloc")]
3233    fn tail(self, n: usize) -> VecDequeIntoIter<Self::Item>
3234    where
3235        Self: Sized,
3236    {
3237        match n {
3238            0 => {
3239                self.last();
3240                VecDeque::new()
3241            }
3242            1 => self.last().into_iter().collect(),
3243            _ => {
3244                // Skip the starting part of the iterator if possible.
3245                let (low, _) = self.size_hint();
3246                let mut iter = self.fuse().skip(low.saturating_sub(n));
3247                // TODO: If VecDeque has a more efficient method than
3248                // `.pop_front();.push_back(val)` in the future then maybe revisit this.
3249                let mut data: Vec<_> = iter.by_ref().take(n).collect();
3250                // Update `data` cyclically.
3251                let idx = iter.fold(0, |i, val| {
3252                    debug_assert_eq!(data.len(), n);
3253                    data[i] = val;
3254                    if i + 1 == n {
3255                        0
3256                    } else {
3257                        i + 1
3258                    }
3259                });
3260                // Respect the insertion order, efficiently.
3261                let mut data = VecDeque::from(data);
3262                data.rotate_left(idx);
3263                data
3264            }
3265        }
3266        .into_iter()
3267    }
3268
3269    /// Collect all iterator elements into one of two
3270    /// partitions. Unlike [`Iterator::partition`], each partition may
3271    /// have a distinct type.
3272    ///
3273    /// ```
3274    /// use itertools::{Itertools, Either};
3275    ///
3276    /// let successes_and_failures = vec![Ok(1), Err(false), Err(true), Ok(2)];
3277    ///
3278    /// let (successes, failures): (Vec<_>, Vec<_>) = successes_and_failures
3279    ///     .into_iter()
3280    ///     .partition_map(|r| {
3281    ///         match r {
3282    ///             Ok(v) => Either::Left(v),
3283    ///             Err(v) => Either::Right(v),
3284    ///         }
3285    ///     });
3286    ///
3287    /// assert_eq!(successes, [1, 2]);
3288    /// assert_eq!(failures, [false, true]);
3289    /// ```
3290    fn partition_map<A, B, F, L, R>(self, mut predicate: F) -> (A, B)
3291    where
3292        Self: Sized,
3293        F: FnMut(Self::Item) -> Either<L, R>,
3294        A: Default + Extend<L>,
3295        B: Default + Extend<R>,
3296    {
3297        let mut left = A::default();
3298        let mut right = B::default();
3299
3300        self.for_each(|val| match predicate(val) {
3301            Either::Left(v) => left.extend(Some(v)),
3302            Either::Right(v) => right.extend(Some(v)),
3303        });
3304
3305        (left, right)
3306    }
3307
3308    /// Partition a sequence of `Result`s into one list of all the `Ok` elements
3309    /// and another list of all the `Err` elements.
3310    ///
3311    /// ```
3312    /// use itertools::Itertools;
3313    ///
3314    /// let successes_and_failures = vec![Ok(1), Err(false), Err(true), Ok(2)];
3315    ///
3316    /// let (successes, failures): (Vec<_>, Vec<_>) = successes_and_failures
3317    ///     .into_iter()
3318    ///     .partition_result();
3319    ///
3320    /// assert_eq!(successes, [1, 2]);
3321    /// assert_eq!(failures, [false, true]);
3322    /// ```
3323    fn partition_result<A, B, T, E>(self) -> (A, B)
3324    where
3325        Self: Iterator<Item = Result<T, E>> + Sized,
3326        A: Default + Extend<T>,
3327        B: Default + Extend<E>,
3328    {
3329        self.partition_map(|r| match r {
3330            Ok(v) => Either::Left(v),
3331            Err(v) => Either::Right(v),
3332        })
3333    }
3334
3335    /// Return a `HashMap` of keys mapped to `Vec`s of values. Keys and values
3336    /// are taken from `(Key, Value)` tuple pairs yielded by the input iterator.
3337    ///
3338    /// Essentially a shorthand for `.into_grouping_map().collect::<Vec<_>>()`.
3339    ///
3340    /// ```
3341    /// use itertools::Itertools;
3342    ///
3343    /// let data = vec![(0, 10), (2, 12), (3, 13), (0, 20), (3, 33), (2, 42)];
3344    /// let lookup = data.into_iter().into_group_map();
3345    ///
3346    /// assert_eq!(lookup[&0], vec![10, 20]);
3347    /// assert_eq!(lookup.get(&1), None);
3348    /// assert_eq!(lookup[&2], vec![12, 42]);
3349    /// assert_eq!(lookup[&3], vec![13, 33]);
3350    /// ```
3351    #[cfg(feature = "use_std")]
3352    fn into_group_map<K, V>(self) -> HashMap<K, Vec<V>>
3353    where
3354        Self: Iterator<Item = (K, V)> + Sized,
3355        K: Hash + Eq,
3356    {
3357        group_map::into_group_map(self)
3358    }
3359
3360    /// Return an `Iterator` on a `HashMap`. Keys mapped to `Vec`s of values. The key is specified
3361    /// in the closure.
3362    ///
3363    /// Essentially a shorthand for `.into_grouping_map_by(f).collect::<Vec<_>>()`.
3364    ///
3365    /// ```
3366    /// use itertools::Itertools;
3367    /// use std::collections::HashMap;
3368    ///
3369    /// let data = vec![(0, 10), (2, 12), (3, 13), (0, 20), (3, 33), (2, 42)];
3370    /// let lookup: HashMap<u32,Vec<(u32, u32)>> =
3371    ///     data.clone().into_iter().into_group_map_by(|a| a.0);
3372    ///
3373    /// assert_eq!(lookup[&0], vec![(0,10),(0,20)]);
3374    /// assert_eq!(lookup.get(&1), None);
3375    /// assert_eq!(lookup[&2], vec![(2,12), (2,42)]);
3376    /// assert_eq!(lookup[&3], vec![(3,13), (3,33)]);
3377    ///
3378    /// assert_eq!(
3379    ///     data.into_iter()
3380    ///         .into_group_map_by(|x| x.0)
3381    ///         .into_iter()
3382    ///         .map(|(key, values)| (key, values.into_iter().fold(0,|acc, (_,v)| acc + v )))
3383    ///         .collect::<HashMap<u32,u32>>()[&0],
3384    ///     30,
3385    /// );
3386    /// ```
3387    #[cfg(feature = "use_std")]
3388    fn into_group_map_by<K, V, F>(self, f: F) -> HashMap<K, Vec<V>>
3389    where
3390        Self: Iterator<Item = V> + Sized,
3391        K: Hash + Eq,
3392        F: FnMut(&V) -> K,
3393    {
3394        group_map::into_group_map_by(self, f)
3395    }
3396
3397    /// Constructs a `GroupingMap` to be used later with one of the efficient
3398    /// group-and-fold operations it allows to perform.
3399    ///
3400    /// The input iterator must yield item in the form of `(K, V)` where the
3401    /// value of type `K` will be used as key to identify the groups and the
3402    /// value of type `V` as value for the folding operation.
3403    ///
3404    /// See [`GroupingMap`] for more informations
3405    /// on what operations are available.
3406    #[cfg(feature = "use_std")]
3407    fn into_grouping_map<K, V>(self) -> GroupingMap<Self>
3408    where
3409        Self: Iterator<Item = (K, V)> + Sized,
3410        K: Hash + Eq,
3411    {
3412        grouping_map::new(self)
3413    }
3414
3415    /// Constructs a `GroupingMap` to be used later with one of the efficient
3416    /// group-and-fold operations it allows to perform.
3417    ///
3418    /// The values from this iterator will be used as values for the folding operation
3419    /// while the keys will be obtained from the values by calling `key_mapper`.
3420    ///
3421    /// See [`GroupingMap`] for more informations
3422    /// on what operations are available.
3423    #[cfg(feature = "use_std")]
3424    fn into_grouping_map_by<K, V, F>(self, key_mapper: F) -> GroupingMapBy<Self, F>
3425    where
3426        Self: Iterator<Item = V> + Sized,
3427        K: Hash + Eq,
3428        F: FnMut(&V) -> K,
3429    {
3430        grouping_map::new(grouping_map::new_map_for_grouping(self, key_mapper))
3431    }
3432
3433    /// Return all minimum elements of an iterator.
3434    ///
3435    /// # Examples
3436    ///
3437    /// ```
3438    /// use itertools::Itertools;
3439    ///
3440    /// let a: [i32; 0] = [];
3441    /// assert_eq!(a.iter().min_set(), Vec::<&i32>::new());
3442    ///
3443    /// let a = [1];
3444    /// assert_eq!(a.iter().min_set(), vec![&1]);
3445    ///
3446    /// let a = [1, 2, 3, 4, 5];
3447    /// assert_eq!(a.iter().min_set(), vec![&1]);
3448    ///
3449    /// let a = [1, 1, 1, 1];
3450    /// assert_eq!(a.iter().min_set(), vec![&1, &1, &1, &1]);
3451    /// ```
3452    ///
3453    /// The elements can be floats but no particular result is guaranteed
3454    /// if an element is NaN.
3455    #[cfg(feature = "use_alloc")]
3456    fn min_set(self) -> Vec<Self::Item>
3457    where
3458        Self: Sized,
3459        Self::Item: Ord,
3460    {
3461        extrema_set::min_set_impl(self, |_| (), |x, y, _, _| x.cmp(y))
3462    }
3463
3464    /// Return all minimum elements of an iterator, as determined by
3465    /// the specified function.
3466    ///
3467    /// # Examples
3468    ///
3469    /// ```
3470    /// # use std::cmp::Ordering;
3471    /// use itertools::Itertools;
3472    ///
3473    /// let a: [(i32, i32); 0] = [];
3474    /// assert_eq!(a.iter().min_set_by(|_, _| Ordering::Equal), Vec::<&(i32, i32)>::new());
3475    ///
3476    /// let a = [(1, 2)];
3477    /// assert_eq!(a.iter().min_set_by(|&&(k1,_), &&(k2, _)| k1.cmp(&k2)), vec![&(1, 2)]);
3478    ///
3479    /// let a = [(1, 2), (2, 2), (3, 9), (4, 8), (5, 9)];
3480    /// assert_eq!(a.iter().min_set_by(|&&(_,k1), &&(_,k2)| k1.cmp(&k2)), vec![&(1, 2), &(2, 2)]);
3481    ///
3482    /// let a = [(1, 2), (1, 3), (1, 4), (1, 5)];
3483    /// assert_eq!(a.iter().min_set_by(|&&(k1,_), &&(k2, _)| k1.cmp(&k2)), vec![&(1, 2), &(1, 3), &(1, 4), &(1, 5)]);
3484    /// ```
3485    ///
3486    /// The elements can be floats but no particular result is guaranteed
3487    /// if an element is NaN.
3488    #[cfg(feature = "use_alloc")]
3489    fn min_set_by<F>(self, mut compare: F) -> Vec<Self::Item>
3490    where
3491        Self: Sized,
3492        F: FnMut(&Self::Item, &Self::Item) -> Ordering,
3493    {
3494        extrema_set::min_set_impl(self, |_| (), |x, y, _, _| compare(x, y))
3495    }
3496
3497    /// Return all minimum elements of an iterator, as determined by
3498    /// the specified function.
3499    ///
3500    /// # Examples
3501    ///
3502    /// ```
3503    /// use itertools::Itertools;
3504    ///
3505    /// let a: [(i32, i32); 0] = [];
3506    /// assert_eq!(a.iter().min_set_by_key(|_| ()), Vec::<&(i32, i32)>::new());
3507    ///
3508    /// let a = [(1, 2)];
3509    /// assert_eq!(a.iter().min_set_by_key(|&&(k,_)| k), vec![&(1, 2)]);
3510    ///
3511    /// let a = [(1, 2), (2, 2), (3, 9), (4, 8), (5, 9)];
3512    /// assert_eq!(a.iter().min_set_by_key(|&&(_, k)| k), vec![&(1, 2), &(2, 2)]);
3513    ///
3514    /// let a = [(1, 2), (1, 3), (1, 4), (1, 5)];
3515    /// assert_eq!(a.iter().min_set_by_key(|&&(k, _)| k), vec![&(1, 2), &(1, 3), &(1, 4), &(1, 5)]);
3516    /// ```
3517    ///
3518    /// The elements can be floats but no particular result is guaranteed
3519    /// if an element is NaN.
3520    #[cfg(feature = "use_alloc")]
3521    fn min_set_by_key<K, F>(self, key: F) -> Vec<Self::Item>
3522    where
3523        Self: Sized,
3524        K: Ord,
3525        F: FnMut(&Self::Item) -> K,
3526    {
3527        extrema_set::min_set_impl(self, key, |_, _, kx, ky| kx.cmp(ky))
3528    }
3529
3530    /// Return all maximum elements of an iterator.
3531    ///
3532    /// # Examples
3533    ///
3534    /// ```
3535    /// use itertools::Itertools;
3536    ///
3537    /// let a: [i32; 0] = [];
3538    /// assert_eq!(a.iter().max_set(), Vec::<&i32>::new());
3539    ///
3540    /// let a = [1];
3541    /// assert_eq!(a.iter().max_set(), vec![&1]);
3542    ///
3543    /// let a = [1, 2, 3, 4, 5];
3544    /// assert_eq!(a.iter().max_set(), vec![&5]);
3545    ///
3546    /// let a = [1, 1, 1, 1];
3547    /// assert_eq!(a.iter().max_set(), vec![&1, &1, &1, &1]);
3548    /// ```
3549    ///
3550    /// The elements can be floats but no particular result is guaranteed
3551    /// if an element is NaN.
3552    #[cfg(feature = "use_alloc")]
3553    fn max_set(self) -> Vec<Self::Item>
3554    where
3555        Self: Sized,
3556        Self::Item: Ord,
3557    {
3558        extrema_set::max_set_impl(self, |_| (), |x, y, _, _| x.cmp(y))
3559    }
3560
3561    /// Return all maximum elements of an iterator, as determined by
3562    /// the specified function.
3563    ///
3564    /// # Examples
3565    ///
3566    /// ```
3567    /// # use std::cmp::Ordering;
3568    /// use itertools::Itertools;
3569    ///
3570    /// let a: [(i32, i32); 0] = [];
3571    /// assert_eq!(a.iter().max_set_by(|_, _| Ordering::Equal), Vec::<&(i32, i32)>::new());
3572    ///
3573    /// let a = [(1, 2)];
3574    /// assert_eq!(a.iter().max_set_by(|&&(k1,_), &&(k2, _)| k1.cmp(&k2)), vec![&(1, 2)]);
3575    ///
3576    /// let a = [(1, 2), (2, 2), (3, 9), (4, 8), (5, 9)];
3577    /// assert_eq!(a.iter().max_set_by(|&&(_,k1), &&(_,k2)| k1.cmp(&k2)), vec![&(3, 9), &(5, 9)]);
3578    ///
3579    /// let a = [(1, 2), (1, 3), (1, 4), (1, 5)];
3580    /// assert_eq!(a.iter().max_set_by(|&&(k1,_), &&(k2, _)| k1.cmp(&k2)), vec![&(1, 2), &(1, 3), &(1, 4), &(1, 5)]);
3581    /// ```
3582    ///
3583    /// The elements can be floats but no particular result is guaranteed
3584    /// if an element is NaN.
3585    #[cfg(feature = "use_alloc")]
3586    fn max_set_by<F>(self, mut compare: F) -> Vec<Self::Item>
3587    where
3588        Self: Sized,
3589        F: FnMut(&Self::Item, &Self::Item) -> Ordering,
3590    {
3591        extrema_set::max_set_impl(self, |_| (), |x, y, _, _| compare(x, y))
3592    }
3593
3594    /// Return all maximum elements of an iterator, as determined by
3595    /// the specified function.
3596    ///
3597    /// # Examples
3598    ///
3599    /// ```
3600    /// use itertools::Itertools;
3601    ///
3602    /// let a: [(i32, i32); 0] = [];
3603    /// assert_eq!(a.iter().max_set_by_key(|_| ()), Vec::<&(i32, i32)>::new());
3604    ///
3605    /// let a = [(1, 2)];
3606    /// assert_eq!(a.iter().max_set_by_key(|&&(k,_)| k), vec![&(1, 2)]);
3607    ///
3608    /// let a = [(1, 2), (2, 2), (3, 9), (4, 8), (5, 9)];
3609    /// assert_eq!(a.iter().max_set_by_key(|&&(_, k)| k), vec![&(3, 9), &(5, 9)]);
3610    ///
3611    /// let a = [(1, 2), (1, 3), (1, 4), (1, 5)];
3612    /// assert_eq!(a.iter().max_set_by_key(|&&(k, _)| k), vec![&(1, 2), &(1, 3), &(1, 4), &(1, 5)]);
3613    /// ```
3614    ///
3615    /// The elements can be floats but no particular result is guaranteed
3616    /// if an element is NaN.
3617    #[cfg(feature = "use_alloc")]
3618    fn max_set_by_key<K, F>(self, key: F) -> Vec<Self::Item>
3619    where
3620        Self: Sized,
3621        K: Ord,
3622        F: FnMut(&Self::Item) -> K,
3623    {
3624        extrema_set::max_set_impl(self, key, |_, _, kx, ky| kx.cmp(ky))
3625    }
3626
3627    /// Return the minimum and maximum elements in the iterator.
3628    ///
3629    /// The return type `MinMaxResult` is an enum of three variants:
3630    ///
3631    /// - `NoElements` if the iterator is empty.
3632    /// - `OneElement(x)` if the iterator has exactly one element.
3633    /// - `MinMax(x, y)` is returned otherwise, where `x <= y`. Two
3634    ///    values are equal if and only if there is more than one
3635    ///    element in the iterator and all elements are equal.
3636    ///
3637    /// On an iterator of length `n`, `minmax` does `1.5 * n` comparisons,
3638    /// and so is faster than calling `min` and `max` separately which does
3639    /// `2 * n` comparisons.
3640    ///
3641    /// # Examples
3642    ///
3643    /// ```
3644    /// use itertools::Itertools;
3645    /// use itertools::MinMaxResult::{NoElements, OneElement, MinMax};
3646    ///
3647    /// let a: [i32; 0] = [];
3648    /// assert_eq!(a.iter().minmax(), NoElements);
3649    ///
3650    /// let a = [1];
3651    /// assert_eq!(a.iter().minmax(), OneElement(&1));
3652    ///
3653    /// let a = [1, 2, 3, 4, 5];
3654    /// assert_eq!(a.iter().minmax(), MinMax(&1, &5));
3655    ///
3656    /// let a = [1, 1, 1, 1];
3657    /// assert_eq!(a.iter().minmax(), MinMax(&1, &1));
3658    /// ```
3659    ///
3660    /// The elements can be floats but no particular result is guaranteed
3661    /// if an element is NaN.
3662    fn minmax(self) -> MinMaxResult<Self::Item>
3663    where
3664        Self: Sized,
3665        Self::Item: PartialOrd,
3666    {
3667        minmax::minmax_impl(self, |_| (), |x, y, _, _| x < y)
3668    }
3669
3670    /// Return the minimum and maximum element of an iterator, as determined by
3671    /// the specified function.
3672    ///
3673    /// The return value is a variant of [`MinMaxResult`] like for [`.minmax()`](Itertools::minmax).
3674    ///
3675    /// For the minimum, the first minimal element is returned.  For the maximum,
3676    /// the last maximal element wins.  This matches the behavior of the standard
3677    /// [`Iterator::min`] and [`Iterator::max`] methods.
3678    ///
3679    /// The keys can be floats but no particular result is guaranteed
3680    /// if a key is NaN.
3681    fn minmax_by_key<K, F>(self, key: F) -> MinMaxResult<Self::Item>
3682    where
3683        Self: Sized,
3684        K: PartialOrd,
3685        F: FnMut(&Self::Item) -> K,
3686    {
3687        minmax::minmax_impl(self, key, |_, _, xk, yk| xk < yk)
3688    }
3689
3690    /// Return the minimum and maximum element of an iterator, as determined by
3691    /// the specified comparison function.
3692    ///
3693    /// The return value is a variant of [`MinMaxResult`] like for [`.minmax()`](Itertools::minmax).
3694    ///
3695    /// For the minimum, the first minimal element is returned.  For the maximum,
3696    /// the last maximal element wins.  This matches the behavior of the standard
3697    /// [`Iterator::min`] and [`Iterator::max`] methods.
3698    fn minmax_by<F>(self, mut compare: F) -> MinMaxResult<Self::Item>
3699    where
3700        Self: Sized,
3701        F: FnMut(&Self::Item, &Self::Item) -> Ordering,
3702    {
3703        minmax::minmax_impl(self, |_| (), |x, y, _, _| Ordering::Less == compare(x, y))
3704    }
3705
3706    /// Return the position of the maximum element in the iterator.
3707    ///
3708    /// If several elements are equally maximum, the position of the
3709    /// last of them is returned.
3710    ///
3711    /// # Examples
3712    ///
3713    /// ```
3714    /// use itertools::Itertools;
3715    ///
3716    /// let a: [i32; 0] = [];
3717    /// assert_eq!(a.iter().position_max(), None);
3718    ///
3719    /// let a = [-3, 0, 1, 5, -10];
3720    /// assert_eq!(a.iter().position_max(), Some(3));
3721    ///
3722    /// let a = [1, 1, -1, -1];
3723    /// assert_eq!(a.iter().position_max(), Some(1));
3724    /// ```
3725    fn position_max(self) -> Option<usize>
3726    where
3727        Self: Sized,
3728        Self::Item: Ord,
3729    {
3730        self.enumerate()
3731            .max_by(|x, y| Ord::cmp(&x.1, &y.1))
3732            .map(|x| x.0)
3733    }
3734
3735    /// Return the position of the maximum element in the iterator, as
3736    /// determined by the specified function.
3737    ///
3738    /// If several elements are equally maximum, the position of the
3739    /// last of them is returned.
3740    ///
3741    /// # Examples
3742    ///
3743    /// ```
3744    /// use itertools::Itertools;
3745    ///
3746    /// let a: [i32; 0] = [];
3747    /// assert_eq!(a.iter().position_max_by_key(|x| x.abs()), None);
3748    ///
3749    /// let a = [-3_i32, 0, 1, 5, -10];
3750    /// assert_eq!(a.iter().position_max_by_key(|x| x.abs()), Some(4));
3751    ///
3752    /// let a = [1_i32, 1, -1, -1];
3753    /// assert_eq!(a.iter().position_max_by_key(|x| x.abs()), Some(3));
3754    /// ```
3755    fn position_max_by_key<K, F>(self, mut key: F) -> Option<usize>
3756    where
3757        Self: Sized,
3758        K: Ord,
3759        F: FnMut(&Self::Item) -> K,
3760    {
3761        self.enumerate()
3762            .max_by(|x, y| Ord::cmp(&key(&x.1), &key(&y.1)))
3763            .map(|x| x.0)
3764    }
3765
3766    /// Return the position of the maximum element in the iterator, as
3767    /// determined by the specified comparison function.
3768    ///
3769    /// If several elements are equally maximum, the position of the
3770    /// last of them is returned.
3771    ///
3772    /// # Examples
3773    ///
3774    /// ```
3775    /// use itertools::Itertools;
3776    ///
3777    /// let a: [i32; 0] = [];
3778    /// assert_eq!(a.iter().position_max_by(|x, y| x.cmp(y)), None);
3779    ///
3780    /// let a = [-3_i32, 0, 1, 5, -10];
3781    /// assert_eq!(a.iter().position_max_by(|x, y| x.cmp(y)), Some(3));
3782    ///
3783    /// let a = [1_i32, 1, -1, -1];
3784    /// assert_eq!(a.iter().position_max_by(|x, y| x.cmp(y)), Some(1));
3785    /// ```
3786    fn position_max_by<F>(self, mut compare: F) -> Option<usize>
3787    where
3788        Self: Sized,
3789        F: FnMut(&Self::Item, &Self::Item) -> Ordering,
3790    {
3791        self.enumerate()
3792            .max_by(|x, y| compare(&x.1, &y.1))
3793            .map(|x| x.0)
3794    }
3795
3796    /// Return the position of the minimum element in the iterator.
3797    ///
3798    /// If several elements are equally minimum, the position of the
3799    /// first of them is returned.
3800    ///
3801    /// # Examples
3802    ///
3803    /// ```
3804    /// use itertools::Itertools;
3805    ///
3806    /// let a: [i32; 0] = [];
3807    /// assert_eq!(a.iter().position_min(), None);
3808    ///
3809    /// let a = [-3, 0, 1, 5, -10];
3810    /// assert_eq!(a.iter().position_min(), Some(4));
3811    ///
3812    /// let a = [1, 1, -1, -1];
3813    /// assert_eq!(a.iter().position_min(), Some(2));
3814    /// ```
3815    fn position_min(self) -> Option<usize>
3816    where
3817        Self: Sized,
3818        Self::Item: Ord,
3819    {
3820        self.enumerate()
3821            .min_by(|x, y| Ord::cmp(&x.1, &y.1))
3822            .map(|x| x.0)
3823    }
3824
3825    /// Return the position of the minimum element in the iterator, as
3826    /// determined by the specified function.
3827    ///
3828    /// If several elements are equally minimum, the position of the
3829    /// first of them is returned.
3830    ///
3831    /// # Examples
3832    ///
3833    /// ```
3834    /// use itertools::Itertools;
3835    ///
3836    /// let a: [i32; 0] = [];
3837    /// assert_eq!(a.iter().position_min_by_key(|x| x.abs()), None);
3838    ///
3839    /// let a = [-3_i32, 0, 1, 5, -10];
3840    /// assert_eq!(a.iter().position_min_by_key(|x| x.abs()), Some(1));
3841    ///
3842    /// let a = [1_i32, 1, -1, -1];
3843    /// assert_eq!(a.iter().position_min_by_key(|x| x.abs()), Some(0));
3844    /// ```
3845    fn position_min_by_key<K, F>(self, mut key: F) -> Option<usize>
3846    where
3847        Self: Sized,
3848        K: Ord,
3849        F: FnMut(&Self::Item) -> K,
3850    {
3851        self.enumerate()
3852            .min_by(|x, y| Ord::cmp(&key(&x.1), &key(&y.1)))
3853            .map(|x| x.0)
3854    }
3855
3856    /// Return the position of the minimum element in the iterator, as
3857    /// determined by the specified comparison function.
3858    ///
3859    /// If several elements are equally minimum, the position of the
3860    /// first of them is returned.
3861    ///
3862    /// # Examples
3863    ///
3864    /// ```
3865    /// use itertools::Itertools;
3866    ///
3867    /// let a: [i32; 0] = [];
3868    /// assert_eq!(a.iter().position_min_by(|x, y| x.cmp(y)), None);
3869    ///
3870    /// let a = [-3_i32, 0, 1, 5, -10];
3871    /// assert_eq!(a.iter().position_min_by(|x, y| x.cmp(y)), Some(4));
3872    ///
3873    /// let a = [1_i32, 1, -1, -1];
3874    /// assert_eq!(a.iter().position_min_by(|x, y| x.cmp(y)), Some(2));
3875    /// ```
3876    fn position_min_by<F>(self, mut compare: F) -> Option<usize>
3877    where
3878        Self: Sized,
3879        F: FnMut(&Self::Item, &Self::Item) -> Ordering,
3880    {
3881        self.enumerate()
3882            .min_by(|x, y| compare(&x.1, &y.1))
3883            .map(|x| x.0)
3884    }
3885
3886    /// Return the positions of the minimum and maximum elements in
3887    /// the iterator.
3888    ///
3889    /// The return type [`MinMaxResult`] is an enum of three variants:
3890    ///
3891    /// - `NoElements` if the iterator is empty.
3892    /// - `OneElement(xpos)` if the iterator has exactly one element.
3893    /// - `MinMax(xpos, ypos)` is returned otherwise, where the
3894    ///    element at `xpos` ≤ the element at `ypos`. While the
3895    ///    referenced elements themselves may be equal, `xpos` cannot
3896    ///    be equal to `ypos`.
3897    ///
3898    /// On an iterator of length `n`, `position_minmax` does `1.5 * n`
3899    /// comparisons, and so is faster than calling `position_min` and
3900    /// `position_max` separately which does `2 * n` comparisons.
3901    ///
3902    /// For the minimum, if several elements are equally minimum, the
3903    /// position of the first of them is returned. For the maximum, if
3904    /// several elements are equally maximum, the position of the last
3905    /// of them is returned.
3906    ///
3907    /// The elements can be floats but no particular result is
3908    /// guaranteed if an element is NaN.
3909    ///
3910    /// # Examples
3911    ///
3912    /// ```
3913    /// use itertools::Itertools;
3914    /// use itertools::MinMaxResult::{NoElements, OneElement, MinMax};
3915    ///
3916    /// let a: [i32; 0] = [];
3917    /// assert_eq!(a.iter().position_minmax(), NoElements);
3918    ///
3919    /// let a = [10];
3920    /// assert_eq!(a.iter().position_minmax(), OneElement(0));
3921    ///
3922    /// let a = [-3, 0, 1, 5, -10];
3923    /// assert_eq!(a.iter().position_minmax(), MinMax(4, 3));
3924    ///
3925    /// let a = [1, 1, -1, -1];
3926    /// assert_eq!(a.iter().position_minmax(), MinMax(2, 1));
3927    /// ```
3928    fn position_minmax(self) -> MinMaxResult<usize>
3929    where
3930        Self: Sized,
3931        Self::Item: PartialOrd,
3932    {
3933        use crate::MinMaxResult::{MinMax, NoElements, OneElement};
3934        match minmax::minmax_impl(self.enumerate(), |_| (), |x, y, _, _| x.1 < y.1) {
3935            NoElements => NoElements,
3936            OneElement(x) => OneElement(x.0),
3937            MinMax(x, y) => MinMax(x.0, y.0),
3938        }
3939    }
3940
3941    /// Return the postions of the minimum and maximum elements of an
3942    /// iterator, as determined by the specified function.
3943    ///
3944    /// The return value is a variant of [`MinMaxResult`] like for
3945    /// [`position_minmax`].
3946    ///
3947    /// For the minimum, if several elements are equally minimum, the
3948    /// position of the first of them is returned. For the maximum, if
3949    /// several elements are equally maximum, the position of the last
3950    /// of them is returned.
3951    ///
3952    /// The keys can be floats but no particular result is guaranteed
3953    /// if a key is NaN.
3954    ///
3955    /// # Examples
3956    ///
3957    /// ```
3958    /// use itertools::Itertools;
3959    /// use itertools::MinMaxResult::{NoElements, OneElement, MinMax};
3960    ///
3961    /// let a: [i32; 0] = [];
3962    /// assert_eq!(a.iter().position_minmax_by_key(|x| x.abs()), NoElements);
3963    ///
3964    /// let a = [10_i32];
3965    /// assert_eq!(a.iter().position_minmax_by_key(|x| x.abs()), OneElement(0));
3966    ///
3967    /// let a = [-3_i32, 0, 1, 5, -10];
3968    /// assert_eq!(a.iter().position_minmax_by_key(|x| x.abs()), MinMax(1, 4));
3969    ///
3970    /// let a = [1_i32, 1, -1, -1];
3971    /// assert_eq!(a.iter().position_minmax_by_key(|x| x.abs()), MinMax(0, 3));
3972    /// ```
3973    ///
3974    /// [`position_minmax`]: Self::position_minmax
3975    fn position_minmax_by_key<K, F>(self, mut key: F) -> MinMaxResult<usize>
3976    where
3977        Self: Sized,
3978        K: PartialOrd,
3979        F: FnMut(&Self::Item) -> K,
3980    {
3981        use crate::MinMaxResult::{MinMax, NoElements, OneElement};
3982        match self.enumerate().minmax_by_key(|e| key(&e.1)) {
3983            NoElements => NoElements,
3984            OneElement(x) => OneElement(x.0),
3985            MinMax(x, y) => MinMax(x.0, y.0),
3986        }
3987    }
3988
3989    /// Return the postions of the minimum and maximum elements of an
3990    /// iterator, as determined by the specified comparison function.
3991    ///
3992    /// The return value is a variant of [`MinMaxResult`] like for
3993    /// [`position_minmax`].
3994    ///
3995    /// For the minimum, if several elements are equally minimum, the
3996    /// position of the first of them is returned. For the maximum, if
3997    /// several elements are equally maximum, the position of the last
3998    /// of them is returned.
3999    ///
4000    /// # Examples
4001    ///
4002    /// ```
4003    /// use itertools::Itertools;
4004    /// use itertools::MinMaxResult::{NoElements, OneElement, MinMax};
4005    ///
4006    /// let a: [i32; 0] = [];
4007    /// assert_eq!(a.iter().position_minmax_by(|x, y| x.cmp(y)), NoElements);
4008    ///
4009    /// let a = [10_i32];
4010    /// assert_eq!(a.iter().position_minmax_by(|x, y| x.cmp(y)), OneElement(0));
4011    ///
4012    /// let a = [-3_i32, 0, 1, 5, -10];
4013    /// assert_eq!(a.iter().position_minmax_by(|x, y| x.cmp(y)), MinMax(4, 3));
4014    ///
4015    /// let a = [1_i32, 1, -1, -1];
4016    /// assert_eq!(a.iter().position_minmax_by(|x, y| x.cmp(y)), MinMax(2, 1));
4017    /// ```
4018    ///
4019    /// [`position_minmax`]: Self::position_minmax
4020    fn position_minmax_by<F>(self, mut compare: F) -> MinMaxResult<usize>
4021    where
4022        Self: Sized,
4023        F: FnMut(&Self::Item, &Self::Item) -> Ordering,
4024    {
4025        use crate::MinMaxResult::{MinMax, NoElements, OneElement};
4026        match self.enumerate().minmax_by(|x, y| compare(&x.1, &y.1)) {
4027            NoElements => NoElements,
4028            OneElement(x) => OneElement(x.0),
4029            MinMax(x, y) => MinMax(x.0, y.0),
4030        }
4031    }
4032
4033    /// If the iterator yields exactly one element, that element will be returned, otherwise
4034    /// an error will be returned containing an iterator that has the same output as the input
4035    /// iterator.
4036    ///
4037    /// This provides an additional layer of validation over just calling `Iterator::next()`.
4038    /// If your assumption that there should only be one element yielded is false this provides
4039    /// the opportunity to detect and handle that, preventing errors at a distance.
4040    ///
4041    /// # Examples
4042    /// ```
4043    /// use itertools::Itertools;
4044    ///
4045    /// assert_eq!((0..10).filter(|&x| x == 2).exactly_one().unwrap(), 2);
4046    /// assert!((0..10).filter(|&x| x > 1 && x < 4).exactly_one().unwrap_err().eq(2..4));
4047    /// assert!((0..10).filter(|&x| x > 1 && x < 5).exactly_one().unwrap_err().eq(2..5));
4048    /// assert!((0..10).filter(|&_| false).exactly_one().unwrap_err().eq(0..0));
4049    /// ```
4050    fn exactly_one(mut self) -> Result<Self::Item, ExactlyOneError<Self>>
4051    where
4052        Self: Sized,
4053    {
4054        match self.next() {
4055            Some(first) => match self.next() {
4056                Some(second) => Err(ExactlyOneError::new(
4057                    Some(Either::Left([first, second])),
4058                    self,
4059                )),
4060                None => Ok(first),
4061            },
4062            None => Err(ExactlyOneError::new(None, self)),
4063        }
4064    }
4065
4066    /// If the iterator yields no elements, `Ok(None)` will be returned. If the iterator yields
4067    /// exactly one element, that element will be returned, otherwise an error will be returned
4068    /// containing an iterator that has the same output as the input iterator.
4069    ///
4070    /// This provides an additional layer of validation over just calling `Iterator::next()`.
4071    /// If your assumption that there should be at most one element yielded is false this provides
4072    /// the opportunity to detect and handle that, preventing errors at a distance.
4073    ///
4074    /// # Examples
4075    /// ```
4076    /// use itertools::Itertools;
4077    ///
4078    /// assert_eq!((0..10).filter(|&x| x == 2).at_most_one().unwrap(), Some(2));
4079    /// assert!((0..10).filter(|&x| x > 1 && x < 4).at_most_one().unwrap_err().eq(2..4));
4080    /// assert!((0..10).filter(|&x| x > 1 && x < 5).at_most_one().unwrap_err().eq(2..5));
4081    /// assert_eq!((0..10).filter(|&_| false).at_most_one().unwrap(), None);
4082    /// ```
4083    fn at_most_one(mut self) -> Result<Option<Self::Item>, ExactlyOneError<Self>>
4084    where
4085        Self: Sized,
4086    {
4087        match self.next() {
4088            Some(first) => match self.next() {
4089                Some(second) => Err(ExactlyOneError::new(
4090                    Some(Either::Left([first, second])),
4091                    self,
4092                )),
4093                None => Ok(Some(first)),
4094            },
4095            None => Ok(None),
4096        }
4097    }
4098
4099    /// An iterator adaptor that allows the user to peek at multiple `.next()`
4100    /// values without advancing the base iterator.
4101    ///
4102    /// # Examples
4103    /// ```
4104    /// use itertools::Itertools;
4105    ///
4106    /// let mut iter = (0..10).multipeek();
4107    /// assert_eq!(iter.peek(), Some(&0));
4108    /// assert_eq!(iter.peek(), Some(&1));
4109    /// assert_eq!(iter.peek(), Some(&2));
4110    /// assert_eq!(iter.next(), Some(0));
4111    /// assert_eq!(iter.peek(), Some(&1));
4112    /// ```
4113    #[cfg(feature = "use_alloc")]
4114    fn multipeek(self) -> MultiPeek<Self>
4115    where
4116        Self: Sized,
4117    {
4118        multipeek_impl::multipeek(self)
4119    }
4120
4121    /// Collect the items in this iterator and return a `HashMap` which
4122    /// contains each item that appears in the iterator and the number
4123    /// of times it appears.
4124    ///
4125    /// # Examples
4126    /// ```
4127    /// # use itertools::Itertools;
4128    /// let counts = [1, 1, 1, 3, 3, 5].into_iter().counts();
4129    /// assert_eq!(counts[&1], 3);
4130    /// assert_eq!(counts[&3], 2);
4131    /// assert_eq!(counts[&5], 1);
4132    /// assert_eq!(counts.get(&0), None);
4133    /// ```
4134    #[cfg(feature = "use_std")]
4135    fn counts(self) -> HashMap<Self::Item, usize>
4136    where
4137        Self: Sized,
4138        Self::Item: Eq + Hash,
4139    {
4140        let mut counts = HashMap::new();
4141        self.for_each(|item| *counts.entry(item).or_default() += 1);
4142        counts
4143    }
4144
4145    /// Collect the items in this iterator and return a `HashMap` which
4146    /// contains each item that appears in the iterator and the number
4147    /// of times it appears,
4148    /// determining identity using a keying function.
4149    ///
4150    /// ```
4151    /// # use itertools::Itertools;
4152    /// struct Character {
4153    ///   first_name: &'static str,
4154    ///   last_name:  &'static str,
4155    /// }
4156    ///
4157    /// let characters =
4158    ///     vec![
4159    ///         Character { first_name: "Amy",   last_name: "Pond"      },
4160    ///         Character { first_name: "Amy",   last_name: "Wong"      },
4161    ///         Character { first_name: "Amy",   last_name: "Santiago"  },
4162    ///         Character { first_name: "James", last_name: "Bond"      },
4163    ///         Character { first_name: "James", last_name: "Sullivan"  },
4164    ///         Character { first_name: "James", last_name: "Norington" },
4165    ///         Character { first_name: "James", last_name: "Kirk"      },
4166    ///     ];
4167    ///
4168    /// let first_name_frequency =
4169    ///     characters
4170    ///         .into_iter()
4171    ///         .counts_by(|c| c.first_name);
4172    ///
4173    /// assert_eq!(first_name_frequency["Amy"], 3);
4174    /// assert_eq!(first_name_frequency["James"], 4);
4175    /// assert_eq!(first_name_frequency.contains_key("Asha"), false);
4176    /// ```
4177    #[cfg(feature = "use_std")]
4178    fn counts_by<K, F>(self, f: F) -> HashMap<K, usize>
4179    where
4180        Self: Sized,
4181        K: Eq + Hash,
4182        F: FnMut(Self::Item) -> K,
4183    {
4184        self.map(f).counts()
4185    }
4186
4187    /// Converts an iterator of tuples into a tuple of containers.
4188    ///
4189    /// It consumes an entire iterator of n-ary tuples, producing `n` collections, one for each
4190    /// column.
4191    ///
4192    /// This function is, in some sense, the opposite of [`multizip`].
4193    ///
4194    /// ```
4195    /// use itertools::Itertools;
4196    ///
4197    /// let inputs = vec![(1, 2, 3), (4, 5, 6), (7, 8, 9)];
4198    ///
4199    /// let (a, b, c): (Vec<_>, Vec<_>, Vec<_>) = inputs
4200    ///     .into_iter()
4201    ///     .multiunzip();
4202    ///
4203    /// assert_eq!(a, vec![1, 4, 7]);
4204    /// assert_eq!(b, vec![2, 5, 8]);
4205    /// assert_eq!(c, vec![3, 6, 9]);
4206    /// ```
4207    fn multiunzip<FromI>(self) -> FromI
4208    where
4209        Self: Sized + MultiUnzip<FromI>,
4210    {
4211        MultiUnzip::multiunzip(self)
4212    }
4213
4214    /// Returns the length of the iterator if one exists.
4215    /// Otherwise return `self.size_hint()`.
4216    ///
4217    /// Fallible [`ExactSizeIterator::len`].
4218    ///
4219    /// Inherits guarantees and restrictions from [`Iterator::size_hint`].
4220    ///
4221    /// ```
4222    /// use itertools::Itertools;
4223    ///
4224    /// assert_eq!([0; 10].iter().try_len(), Ok(10));
4225    /// assert_eq!((10..15).try_len(), Ok(5));
4226    /// assert_eq!((15..10).try_len(), Ok(0));
4227    /// assert_eq!((10..).try_len(), Err((usize::MAX, None)));
4228    /// assert_eq!((10..15).filter(|x| x % 2 == 0).try_len(), Err((0, Some(5))));
4229    /// ```
4230    fn try_len(&self) -> Result<usize, size_hint::SizeHint> {
4231        let sh = self.size_hint();
4232        match sh {
4233            (lo, Some(hi)) if lo == hi => Ok(lo),
4234            _ => Err(sh),
4235        }
4236    }
4237}
4238
4239impl<T> Itertools for T where T: Iterator + ?Sized {}
4240
4241/// Return `true` if both iterables produce equal sequences
4242/// (elements pairwise equal and sequences of the same length),
4243/// `false` otherwise.
4244///
4245/// [`IntoIterator`] enabled version of [`Iterator::eq`].
4246///
4247/// ```
4248/// assert!(itertools::equal(vec![1, 2, 3], 1..4));
4249/// assert!(!itertools::equal(&[0, 0], &[0, 0, 0]));
4250/// ```
4251pub fn equal<I, J>(a: I, b: J) -> bool
4252where
4253    I: IntoIterator,
4254    J: IntoIterator,
4255    I::Item: PartialEq<J::Item>,
4256{
4257    a.into_iter().eq(b)
4258}
4259
4260/// Assert that two iterables produce equal sequences, with the same
4261/// semantics as [`equal(a, b)`](equal).
4262///
4263/// **Panics** on assertion failure with a message that shows the
4264/// two different elements and the iteration index.
4265///
4266/// ```should_panic
4267/// # use itertools::assert_equal;
4268/// assert_equal("exceed".split('c'), "excess".split('c'));
4269/// // ^PANIC: panicked at 'Failed assertion Some("eed") == Some("ess") for iteration 1'.
4270/// ```
4271pub fn assert_equal<I, J>(a: I, b: J)
4272where
4273    I: IntoIterator,
4274    J: IntoIterator,
4275    I::Item: fmt::Debug + PartialEq<J::Item>,
4276    J::Item: fmt::Debug,
4277{
4278    let mut ia = a.into_iter();
4279    let mut ib = b.into_iter();
4280    let mut i: usize = 0;
4281    loop {
4282        match (ia.next(), ib.next()) {
4283            (None, None) => return,
4284            (a, b) => {
4285                let equal = match (&a, &b) {
4286                    (Some(a), Some(b)) => a == b,
4287                    _ => false,
4288                };
4289                assert!(
4290                    equal,
4291                    "Failed assertion {a:?} == {b:?} for iteration {i}",
4292                    i = i,
4293                    a = a,
4294                    b = b
4295                );
4296                i += 1;
4297            }
4298        }
4299    }
4300}
4301
4302/// Partition a sequence using predicate `pred` so that elements
4303/// that map to `true` are placed before elements which map to `false`.
4304///
4305/// The order within the partitions is arbitrary.
4306///
4307/// Return the index of the split point.
4308///
4309/// ```
4310/// use itertools::partition;
4311///
4312/// # // use repeated numbers to not promise any ordering
4313/// let mut data = [7, 1, 1, 7, 1, 1, 7];
4314/// let split_index = partition(&mut data, |elt| *elt >= 3);
4315///
4316/// assert_eq!(data, [7, 7, 7, 1, 1, 1, 1]);
4317/// assert_eq!(split_index, 3);
4318/// ```
4319pub fn partition<'a, A: 'a, I, F>(iter: I, mut pred: F) -> usize
4320where
4321    I: IntoIterator<Item = &'a mut A>,
4322    I::IntoIter: DoubleEndedIterator,
4323    F: FnMut(&A) -> bool,
4324{
4325    let mut split_index = 0;
4326    let mut iter = iter.into_iter();
4327    while let Some(front) = iter.next() {
4328        if !pred(front) {
4329            match iter.rfind(|back| pred(back)) {
4330                Some(back) => std::mem::swap(front, back),
4331                None => break,
4332            }
4333        }
4334        split_index += 1;
4335    }
4336    split_index
4337}
4338
4339/// An enum used for controlling the execution of `fold_while`.
4340///
4341/// See [`.fold_while()`](Itertools::fold_while) for more information.
4342#[derive(Copy, Clone, Debug, Eq, PartialEq)]
4343pub enum FoldWhile<T> {
4344    /// Continue folding with this value
4345    Continue(T),
4346    /// Fold is complete and will return this value
4347    Done(T),
4348}
4349
4350impl<T> FoldWhile<T> {
4351    /// Return the value in the continue or done.
4352    pub fn into_inner(self) -> T {
4353        match self {
4354            Self::Continue(x) | Self::Done(x) => x,
4355        }
4356    }
4357
4358    /// Return true if `self` is `Done`, false if it is `Continue`.
4359    pub fn is_done(&self) -> bool {
4360        match *self {
4361            Self::Continue(_) => false,
4362            Self::Done(_) => true,
4363        }
4364    }
4365}