如何迭代包含相等元素的切片的连续子序列？

Question

我有一个类型的元素序列，它在切片中实现

PartialEq

。为了便于说明，我们假设它看起来像这样：

let data = [1,1,1,2,2,3,4,5,5,5,5,6];

我想迭代该序列的借用切片，以便这些切片的所有元素都按照

PartialEq

相等。例如，在上面的切片中

data

我想要一个迭代器，它会产生：

&data[0..3]   // [1,1,1]
&data[3..5]   // [2,2]
&data[5..6]   // [3]
&data[6..7]   // [4]
&data[7..11]  // [5,5,5,5]
&data[11..12] // [6]

看起来 slice::group_by 正是我所需要的，但从 Rust 1.72.0 开始，它还不稳定。是否有任何直接的方法可以通过使用第 3 方板条箱或结合使用稳定的 std lib API 以稳定的方式获得此功能？

Answer 1

group_by

是一个非常简单的函数。它返回一个具有 this

next

实现的迭代器：

fn next(&mut self) -> Option<Self::Item> {
    if self.slice.is_empty() {
        None
    } else {
        let mut len = 1;
        let mut iter = self.slice.windows(2);
        while let Some([l, r]) = iter.next() {
            if (self.predicate)(l, r) { len += 1 } else { break }
        }
        let (head, tail) = self.slice.split_at(len);
        self.slice = tail;
        Some(head)
    }
}

由于您有一个简单的谓词，因此您可以使用

from_fn

创建一个更短的迭代器。

fn group_by<T: PartialEq>(mut slice: &[T]) -> impl Iterator<Item = &[T]> + '_ {
    std::iter::from_fn(move || {
        let first = slice.first()?;
        let len = slice.iter().position(|t| first != t).unwrap_or(slice.len());
        let (head, tail) = slice.split_at(len);
        slice = tail;
        Some(head)
    })
}