如何在Rust迭代器中重复除最后一项之外的克隆操作

Question

我经常看到这样的模式，即我需要在一个范围内循环，并调用一个拥有其参数所有权的函数。由于该函数拥有其参数的所有权，因此我必须克隆该值。例如：

let a = "hello".to_string();
for i in 0..10 {
print_me(a.clone());
}
fn print_me(s: String) {
println!("{}", s);
}

我想找到一种模式，其中除了最后一个之外，所有调用的值都被克隆，以防止冗余克隆。

最简单的方法是在循环索引上添加显式检查：

for i in 0..10 {
if i == 9 {
print_me(a);
} else {
print_me(a.clone());
}
}

但是编译器无法理解模式，并抱怨迭代过程中移动的值。

我还尝试构建一个迭代器，使用repeat_with()、chain()和once()进行克隆和移动，但闭包包含值：

let iter = std::iter::repeat_with(|| a.clone())
.take(9)
.chain(std::iter::once(a))

实现这种模式最简单的方法是什么？

谢谢！

Answer 1

作为一种替代解决方案，您可以通过打破循环来让编译器理解这实际上是最后一次迭代：

for i in 0..10 {
if i == 9 {
print_me(a);
break;
} else {
print_me(a.clone());
}
}

Answer 2

itertools::repeat_n正是这样做的：

#[derive(Clone, Debug)]
pub struct RepeatN<A> {
elt: Option<A>,
n: usize,
}
/// Create an iterator that produces `n` repetitions of `element`.
pub fn repeat_n<A: Clone>(element: A, n: usize) -> RepeatN<A> {
if n == 0 {
RepeatN { elt: None, n, }
} else {
RepeatN { elt: Some(element), n, }
}
}
impl<A: Clone> Iterator for RepeatN<A> {
type Item = A;
fn next(&mut self) -> Option<Self::Item> {
if self.n > 1 {
self.n -= 1;
self.elt.as_ref().cloned()
} else {
self.n = 0;
self.elt.take()
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
(self.n, Some(self.n))
}
}

然而，使用unsafe代码可以使实现(稍微(更高效。

Answer 3

这是一个简单的解决方案，但可能不是最快的
至少它避免了对最后一项的检查

if let Some((last, lead)) = (0..10).collect::<Vec<u16>>().split_last() {
for _ in lead {
print_me(a.clone());
}
print_me(a);  // last
};