如何在 Rust 中获得一个指向最大值元素的迭代器？

Question

我想访问Vec<i32> 中最大元素旁边的元素。我正在寻找这样的东西：

let v = vec![1, 3, 2];
let it = v.iter().max_element();
assert_eq!(Some(&2), it.next());

在 C++ 中，我会使用 std::max_element，然后只增加迭代器（有或没有边界检查，取决于我目前的冒险程度）。 Rust max 只返回对元素的引用，这对我的用例来说还不够好。

我想出的唯一解决方案是使用enumerate 来获取项目的索引 - 但与 C++ 方式相比，这似乎是手动且繁琐的。

我更喜欢标准库中的东西。

这个例子被简化了——我实际上想附加到最高值，然后从那个点循环整个容器（可能使用cycle() 或类似的东西）。

Answer 1

C++ 迭代器与 Rust 迭代器不同。 Rust 迭代器是只向前的，并且只能被遍历一次。 C++ 迭代器可以被认为是游标。有关详细信息，请参阅What are the main differences between a Rust Iterator and C++ Iterator?。

为了以最通用的方式实现您的目标，您必须遍历整个迭代器以找到最大值。在此过程中，每次找到新的最大值时，您都必须复制 迭代器。最后可以返回最大值后点对应的迭代器。

trait MaxElement {
    type Iter;

    fn max_element(self) -> Self::Iter;
}

impl<I> MaxElement for I
where
    I: Iterator + Clone,
    I::Item: PartialOrd,
{
    type Iter = Self;

    fn max_element(mut self) -> Self::Iter {
        let mut max_iter = self.clone();
        let mut max_val = None;

        while let Some(val) = self.next() {
            if max_val.as_ref().map_or(true, |m| &val > m) {
                max_iter = self.clone();
                max_val = Some(val);
            }
        }

        max_iter
    }
}

fn main() {
    let v = vec![1, 3, 2];
    let mut it = v.iter().max_element();
    assert_eq!(Some(&2), it.next());
}

另见：

• How can I add new methods to Iterator?

我实际上想附加到最高值，然后从那个点循环整个容器（可能使用cycle() 或类似的东西）。

在这种情况下，我会尝试更明显：

fn index_of_max(values: &[i32]) -> Option<usize> {
    values
        .iter()
        .enumerate()
        .max_by_key(|(_idx, &val)| val)
        .map(|(idx, _val)| idx)
}

fn main() {
    let v = vec![1, 3, 2];
    let idx = index_of_max(&v).unwrap_or(0);
    let (a, b) = v.split_at(idx);
    let mut it = b.iter().chain(a).skip(1);
    assert_eq!(Some(&2), it.next());
}

另见：

• What's the fastest way of finding the index of the maximum value in an array?
• Using max_by_key on a vector of floats
• What is the idiomatic way to get the index of a maximum or minimum floating point value in a slice or Vec in Rust?
• Find the item in an array with the largest property

Answer 2

如果您想要的只是最大值之后的项目的值，我会通过简单调用 fold 来完成，跟踪目前找到的最大值和相应的下一个值：

fn main() {
    let v = vec![1, 3, 2];
    let nxt = v.iter().fold (
        (None, None),
        |acc, x| {
            match acc {
                (Some (max), _) if x > max => (Some (x), None),
                (Some (max), None) => (Some (max), Some (x)),
                (None, _) => (Some (x), None),
                _ => acc
            }
        }
    ).1;
    assert_eq!(Some(&2), nxt);
}

playground

根据您想对最大值之后的项目做什么，类似的方法可能允许您一次性完成。

Answer 3

一个简单的解决方案是使用折叠， 以下代码产生“最大数为：99”

    let vv:Vec<i32> = (1..100).collect();
    let largest = vv.iter().fold(std::i32::MIN, |a,b| a.max(*b));
    println!("largest {} ", largest);