Rust中遍历元素对并将首个元素更新为第二个的最优方法及代码可行性
Rust中遍历元素对并更新第一个元素的最佳实现方式
首先你给出的代码存在语法错误,Rust里没有let (a, b) in ...这种遍历写法,正确的遍历迭代器的方式是用for循环。
基础实现方式
如果只是要将ass中的每个元素对应替换为bs中相同位置的元素,最直接的写法是:
for (a, b) in ass.iter_mut().zip(bs.iter()) { *a = *b; }
这里需要注意:
ass必须是可变的(比如mut Vec<T>),这样才能调用iter_mut()获取可变引用- 若
T类型实现了Copytrait,上述代码可以直接运行;如果T是不可复制的类型,需要改成*a = b.clone();,或者如果bs之后不再使用,可以直接移动元素:
for (a, b) in ass.iter_mut().zip(bs) { *a = b; }
处理长度不一致的情况
zip会以两个迭代器中较短的那个为准,自动忽略较长的剩余元素。如果你需要确保ass和bs长度完全一致,可以:
- 先断言长度相等,避免静默忽略元素:
assert_eq!(ass.len(), bs.len()); for (a, b) in ass.iter_mut().zip(bs.iter()) { *a = *b; }
- 使用
itertools库的zip_eq方法(需要在Cargo.toml中添加itertools依赖),它会在长度不一致时直接panic:
use itertools::Itertools; for (a, b) in ass.iter_mut().zip_eq(&bs) { *a = *b; }
更高效的替代写法
如果只是要把bs的内容完整复制到ass中,且两者长度相同,还可以直接用切片赋值:
ass.copy_from_slice(&bs);
这个方法效率更高,因为它是底层的内存批量复制(要求T实现Copy),比遍历迭代器的开销更小。
内容的提问来源于stack exchange,提问作者David 天宇 Wong
相关产品推荐
相关产品推荐

