rust使用vec在遍歷時洗掉元素
需求: 作業中有一個地方需要用到在遍歷時把不符合條件的元素洗掉掉，
比如一個vec中是原始資料是
vec![1,2,3,3,4,5]，然后我要在遍歷中把等于c的元素洗掉掉,目的是得到vec![1,2,4,5]
第一次錯誤嘗試
由于最開始只知道移除元素用remove方法，所以最開始是這樣寫的
let mut items:Vec<&str> = vec!["a", "b", "c", "c", "d", "e"];
println!("before items is {:?}", items);
for (index, item) in items.iter().enumerate() {
if *item == "c" {
items.remove(index);
}
}
println!("then items is {:?}", items);
但是報錯了，報錯提示為
error[E0502]: cannot borrow `items` as mutable because it is also borrowed as immutable
--> src/main.rs:8:13
|
6 | for (index, item) in items.iter().enumerate() {
| ------------------------
| |
| immutable borrow occurs here
| immutable borrow later used here
7 | if *item == "c" {
8 | items.remove(index);
| ^^^^^^^^^^^^^^^^^^^ mutable borrow occurs here
For more information about this error, try `rustc --explain E0502`.
提示很明顯，iter()是不可變的參考，但是呼叫remove的時候洗掉一個元素得時候，對items是可變的參考了，所以一個變數不能既是可變參考又是不可變參考，所以報錯了，
第二次錯誤嘗試
然后我進行了第二次嘗試，
代碼改成了，不在遍歷中進行移除，把需要移除的id保存起來，在遍歷結束之后，再把需要移除的元素給移除掉，
let mut items:Vec<&str> = vec!["a", "b", "c", "c", "d", "e"];
println!("before items is {:?}", items);
let mut remove_indexs: Vec<usize> = Vec::new();
for (index, item) in items.iter().enumerate() {
if *item == "c" {
remove_indexs.push(index);
}
}
println!("remove indexs is {:?}", remove_indexs);
for i in remove_indexs {
items.remove(i);
}
println!("then items is {:?}", items);
列印出的結果是
before items is ["a", "b", "c", "c", "d", "e"]
remove indexs is [2, 3]
then items is ["a", "b", "c", "e"]
不報錯了，但是結果不對啊，預計是[“a”,"b","d","e"]的，怎么變成了["a", "b", "c", "e"]了，
原因是:
第二個回圈需要移除索引為2和3的兩個元素，所以有兩次回圈：
- 第一次：在移除
索引為2的元素之后，items變成了[“a”,"b","c","d","e"],后面的元素補上去了，這時我們的目標索引3的index從3變成了2. - 第二次：按照代碼把
索引為3的元素移除，但是索引3的元素卻成c變成了d，這段程式也就錯了，
正確方法
上面方法失敗之后，覺得vec應該有處理這種情況的方法，所以看了原始碼，找到了答案，
vec有兩個方法可以實作我想要的，
vec.retain 和 vec.drain_filter
vec.retain
vec.retain 很簡單，retain的意思是保留，所以這個方法的意思就是接收一個回呼函式，然后回呼函式里面回傳true進行保留，回傳false的就移除，
示例：
let mut vec = vec![1, 2, 3, 4];
vec.retain(|&x| x % 2 == 0);
assert_eq!(vec, [2, 4]);
所以用vec.retain來實作的話，就是這樣
let mut items:Vec<&str> = vec!["a", "b", "c", "c", "d", "e"];
println!("before items is {:?}", items);
items.retain(|item| if *item == "c" {false } else { true });
println!("then items is {:?}", items);
vec.drain_filter
drain的意思是 排出 的意思，所以這個函式就是排出過濾器，接收一個回呼函式，然后把回呼函式里面回傳true的元素就會排出，自然也就從原本的vec里面洗掉掉了，然后有需要的話還可以搜集排出的元素，
示例:
let mut numbers = vec![1, 2, 3, 4, 5, 6, 8, 9, 11, 13, 14, 15];
let evens = numbers.drain_filter(|x| *x % 2 == 0).collect::<Vec<_>>();
let odds = numbers;
assert_eq!(evens, vec![2, 4, 6, 8, 14]);
assert_eq!(odds, vec![1, 3, 5, 9, 11, 13, 15]);
所以用vec.drain_filter來實作的話，就是這樣
let mut items:Vec<&str> = vec!["a", "b", "c", "c", "d", "e"];
println!("before items is {:?}", items);
items.retain(|item| if *item == "c" {false } else { true });
let removed_items = items.drain_filter(|item| if *item == "c" { true} else { false}).collect::<Vec<_>>();
println!("then items is {:?}", items);
println!("removed item is {:?}", removed_items);
但是報錯了，提示：
error[E0658]: use of unstable library feature 'drain_filter': recently added
--> src/main.rs:7:31
|
7 | let removed_items = items.drain_filter(|item| if *item == "c" { true} else { false}).collect::<Vec<_>>();
| ^^^^^^^^^^^^
|
= note: see issue #43244 <https://github.com/rust-lang/rust/issues/43244> for more information
For more information about this error, try `rustc --explain E0658`.
這個函式是屬于不穩定的特性的，所以需要使用的話是有條件的
使用unstable feature的條件和步驟：
- 只有nightly才可以使用unstable
- 找到unstable feature的名字
-
![feature(xxx)]啟用這個feature
感興趣的可以參考：https://blog.csdn.net/varding/article/details/48206689 來啟用
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/431980.html
標籤：其他