微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Rust中是否有一个特征,可以按组折叠而无需像协程这样的实验功能?

如何解决Rust中是否有一个特征,可以按组折叠而无需像协程这样的实验功能?

我想避免实验性功能,因此在生成器之前我正在考虑其他选项。

我有一个向量v,该向量由0分隔成组。可以通过for循环或迭代器将这些分隔的组的总和累加:

fn main() {
    let v = vec![55,23,75,12,34,97,71,0];

    // for loop
    let mut acc = 0;

    for &vi in &v {
        acc += vi;

        if vi == 0 {
            println!("for yield {}",acc);
            acc = 0;
        }
    }

    // iter 'loop'
    v.iter()
        .scan(0,|acc,&vi| {
            *acc += vi;
            Some(if vi == 0 {
                let total = *acc;
                *acc = 0;
                Some(total)
            } else {
                None
            })
        })
        .filter_map(|acc| acc)
        .for_each(|acc| println!("iter yield {:?}",acc));
}

scan用作 ad hoc 协程,当迭代器产生值时返回Some(value),而在迭代器仍在处理时返回NoneNone会被滤除并打印出总和。

上面的例子有些琐碎,因为两个操作都产生相同的结果。但是,我的项目需要消化数据流(无法收集到向量中)并有条件地折叠它们(例如,这里的条件可能是acc被10整除而不是定界为0)。

折叠条件本身是流水线的,因此每个流水线都可以提供一个迭代器(请考虑嵌套的协程)。我想看看Iterator::scan-> Iterator::filter_map是否有替代方法,无论他们是否使用迭代器。请记住,不可能收集整个数据流,因为数据流可能是无限的。

解决方法

for循环或带有scan()的版本都没有问题。如果我正在寻找更清晰的替代方法,则可以考虑使用group_by中的itertools crate。这个版本很优雅,几乎可以满足您的需求:

use itertools::Itertools;

v.iter()
    .group_by(|&&vi| vi != 0)
    .into_iter()
    .map(|(_,group)| group.sum::<u32>())
    .for_each(|s| println!("sum {}",s));

问题在于,它输出数字153、0、46、0、191和0。0源自这样一个事实,即它认为元素0只是键的另一个组(表达式{ {1}}恰好是假的而不是真实的。要解决此问题,您需要将vi != 0更改为map,类似于对filter_map的操作:

scan()

不够完美,但仍比原始表述更加优雅,因为它使分组更加明确。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。