gpt4 book ai didi

parsing - 将迭代器解交织成两个单独的迭代器

转载 作者:行者123 更新时间:2023-12-03 11:34:41 24 4
gpt4 key购买 nike

与此问题类似:How do I interleave two Rust vectors by chunks of threes into a new vector?
相反,我想执行相反的行为。将迭代器分为两个不同的迭代器,而不必收集它们的内容(也就是我不想使用unzip)。
此外,我想分离一个迭代器而不收集原始迭代器的全部内容。
这是一个半示例:

#[derive(Debug)]
struct Foo(i32);

fn main() {
// note array elements of Foo cannot be copied or cloned
// v1: [0, 1, 2, 3]
// v2: [4, 5, 6, 7]
let v1 = (0..4).map(|num| Foo(num)).collect::<Vec<_>>();
let v2 = (4..8).map(|num| Foo(num)).collect::<Vec<_>>();

// generate interleaved iterator
// interleaved: [0, 1, 4, 5, 2, 3, 6, 7]
let interleaved = v1.chunks(2)
.zip(v2.chunks(2))
.flat_map(|(c1, c2)| c1.iter().chain(c2));

println!("interleaved: {:?}", interleaved.collect::<Vec<_>>());

// regenerate interleaved iterator
let interleaved = v1.chunks(2)
.zip(v2.chunks(2))
.flat_map(|(c1, c2)| c1.iter().chain(c2));

let mut v3: Vec<&Foo> = vec![];
let mut v4: Vec<&Foo> = vec![];

for (idx, item) in interleaved.enumerate() {
if idx % 4 < 2 {
v3.push(item);
} else {
v4.push(item);
}
}

println!("v3: {:?}, v4: {:?}", v3, v4);
}
我想找到一种惯用的解决方案来取消对迭代器的交织,而不必将其引用收集到容器中。
更一般而言,我想弄清楚如何将 n元素的迭代器简化为 mn >= m元素。在这种情况下,我想将长度为8的交错式迭代器减少为长度为2的块的迭代器,然后将分块的迭代器分为其他两个迭代器。
从这种意义上讲,我想将迭代器“解析”为2个块。

最佳答案

如果原始迭代器实现Clone,那么您可以克隆它并分别执行enumerate() + filter():

fn uninterleave<T, I: Iterator<Item = T> + Clone>(
iter: I,
) -> (impl Iterator<Item = T>, impl Iterator<Item = T>) {
let iter1 = iter
.clone()
.enumerate()
.filter(|(i, _)| i % 4 < 2)
.map(|(_, value)| value);

let iter2 = iter
.enumerate()
.filter(|(i, _)| i % 4 >= 2)
.map(|(_, value)| value);

(iter1, iter2)
}

fn main() {
let v = vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9];
let iter = v.iter();

let (iter1, iter2) = uninterleave(iter);
for value in iter1 {
print!("{} ", value);
}
println!();
for value in iter2 {
print!("{} ", value);
}
}
打印:
0 1 4 5 8 9 
2 3 6 7

关于parsing - 将迭代器解交织成两个单独的迭代器,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66164084/

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