gpt4 book ai didi

iterator - 如何在不收集到临时向量的情况下对结果的迭代器执行迭代器计算?

转载 作者:行者123 更新时间:2023-11-29 08:02:06 27 4
gpt4 key购买 nike

我正在寻找一种方法来消除此示例中的临时向量分配:

fn doit<T: Iterator<Item = Result<i32, &'static str>>>(name: &str, iter: T) {
println!(
"{}: {:?}",
name,
iter.collect::<Result<Vec<_>, _>>()
.map(|v| v.into_iter().min())
);
}

fn main() {
let without_errors = vec![Ok(1), Ok(2), Ok(3)];
let with_errors = vec![Ok(1), Err("error"), Ok(2)];

doit("without errors", without_errors.into_iter());
doit("with errors", with_errors.into_iter());
}

这是带有错误处理的迭代器主题的一个变体,除了我不想创建一个集合(所以 collect() 不是工作),但我想对被迭代的元素执行进一步的操作。

请注意,这给出了错误的结果,因为 Ok 小于 Err:

fn doit<T: Iterator<Item = Result<i32, &'static str>>>(name: &str, iter: T) {
println!("{}: {:?}", name, iter.min());
}

它会意外地为 max() 给出正确的结果,但它不会在出现第一个错误时停止迭代。

最佳答案

Iterator::try_fold提供您所需的框架,自 Rust 1.27 ( Playground ) 起可用:

fn fold_ok<I, T, E, F>(mut iter: I, f: F) -> Result<Option<T>, E>
where
I: Iterator<Item = Result<T, E>>,
T: Ord,
F: Fn(T, T) -> T,
{
iter.try_fold(None, |r, i| {
let i = i?;
Ok(Some(if let Some(r) = r { f(r, i) } else { i }))
})
}

fn main() {
let without_errors = vec![Ok(1), Ok(2), Ok(3)];
let with_errors = vec![Ok(1), Err("error"), Ok(2)];

fn doit<'r, T>(name: &str, iter: T)
where
T: Iterator<Item = &'r Result<i32, &'static str>> + Clone,
{
println!("{}: {:?}", name, fold_ok(iter.cloned(), ::std::cmp::min));
}

doit("without errors", without_errors.iter());
doit("with errors", with_errors.iter());
}

在此之前,我认为您唯一的选择是手动迭代 ( Playground )

fn fold_ok<I, T, E, F>(mut iter: I, f: F) -> Result<Option<T>, E>
where
I: Iterator<Item = Result<T, E>>,
T: Ord,
F: Fn(T, T) -> T,
{
let mut result = match iter.next() {
None => return Ok(None),
Some(r) => r?,
};

for item in iter {
result = f(result, item?);
}

Ok(Some(result))
}

fn main() {
let without_errors = vec![Ok(1), Ok(2), Ok(3)];
let with_errors = vec![Ok(1), Err("error"), Ok(2)];

fn doit<'r, T>(name: &str, iter: T)
where
T: Iterator<Item = &'r Result<i32, &'static str>> + Clone,
{
println!(
"{}: {:?}",
name,
fold_ok(iter.clone().cloned(), ::std::cmp::min)
);
}

doit("without errors", without_errors.iter());
doit("with errors", with_errors.iter());
}

关于iterator - 如何在不收集到临时向量的情况下对结果的迭代器执行迭代器计算?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48841367/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com