Rust中原地替换向量中指定值元素的最地道实现方式是什么?
在Rust中原地替换向量中所有匹配元素的地道写法
1. 去掉冗余的else分支
你的原代码里的空else分支完全可以省略,Rust允许if语句仅在条件满足时执行逻辑,不满足时默认什么都不做:
let mut v = vec![1, 2, 3, 3, 6, 2]; v.iter_mut().for_each(|x| if *x == 2 { *x = 0 }); assert_eq!(vec![1, 0, 3, 3, 6, 0], v);
2. 使用for循环迭代可变引用
这是Rust中最直观且可读性极强的写法,直接遍历向量的可变引用,遇到匹配元素就修改:
let mut v = vec![1, 2, 3, 3, 6, 2]; for x in &mut v { if *x == 2 { *x = 0; } } assert_eq!(vec![1, 0, 3, 3, 6, 0], v);
3. 先筛选再修改
通过filter方法先过滤出所有需要修改的元素,再统一执行替换操作,逻辑拆分更清晰,适合条件复杂的场景:
let mut v = vec![1, 2, 3, 3, 6, 2]; v.iter_mut() .filter(|x| **x == 2) .for_each(|x| *x = 0); assert_eq!(vec![1, 0, 3, 3, 6, 0], v);
这些写法都是Rust社区认可的地道实现,选择哪种取决于你的代码风格和场景需求——逻辑简单时前两种足够简洁,条件复杂时第三种的拆分逻辑会让代码更易维护。
内容的提问来源于stack exchange,提问作者Michael Hall
相关产品推荐
相关产品推荐

