用Rust刷LeetCode之27 移除元素

27. 移除元素

难度: 简单

原描述:

用Rust刷LeetCode之27 移除元素_第1张图片

新描述:

用Rust刷LeetCode之27 移除元素_第2张图片


func removeElement(nums []int, val int) int {
 for i := 0; i < len(nums); i++ {
  if nums[i] == val {
   nums = append(nums[:i], nums[i+1:]...)
   i--
  }
 }
 return len(nums)

}
用Rust刷LeetCode之27 移除元素_第3张图片

Rust 版本


下面这种写法编译无法通过:

pub fn remove_element(nums: &mut Vec<i32>, val: i32) -> i32 {
    for i in 0..nums.len() {
        if nums[i] == val {
            nums.remove(i);
            i = i - 1;
        }
    }

    return nums.len() as i32;
}

error[E0384]: cannot assign twice to immutable variable i


在 Rust 中,for 循环中的循环变量是不可变的,因此不能直接修改 i 的值。如果想在循环中修改 i 的值,可以改用 while 循环,并手动更新 i 的值。以下是一个示例的实现:

fn main() {
    let mut nums = vec![3141592654];
    let len = remove_element(&mut nums, 4);
    println!("移除元素后vec的长度为: {:?}", len); // 输出: 8
}

pub fn remove_element(nums: &mut Vec<i32>, val: i32) -> i32 {
    let mut i = 0;

    while i < nums.len() {
        if nums[i] == val {
            nums.remove(i);
        } else {
            i += 1;
        }
    }

    nums.len() as i32
}
用Rust刷LeetCode之27 移除元素_第4张图片

本文由 mdnice 多平台发布

你可能感兴趣的:(后端)