移除元素 27

6 阅读3分钟

力扣 27 移除元素

给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

假设 nums 中不等于 val 的元素数量为 k,要通过此题,您需要执行以下操作:

  • 更改 nums 数组,使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
  • 返回 k

示例 1: **输入:**nums = [3,2,2,3], val = 3 **输出:**2, nums = [2,2,,] **解释:**你的函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。 你在返回的 k 个元素之外留下了什么并不重要(因此它们并不计入评测)。

示例 2: **输入:**nums = [0,1,2,2,3,0,4,2], val = 2 **输出:**5, nums = [0,1,4,0,3,,,_] **解释:**你的函数应该返回 k = 5,并且 nums 中的前五个元素为 0,0,1,3,4。 注意这五个元素可以任意顺序返回。 你在返回的 k 个元素之外留下了什么并不重要(因此它们并不计入评测)。

提示:

  • 0 <= nums.length <= 100
  • 0 <= nums[i] <= 50
  • 0 <= val <= 100

方法一 暴力解法

直接走双层循环,第一层循环遍历数组,判断数组元素与目标值是否相等。如果相等,开始第二层循环,将后方元素前移覆盖目标值位置,实现元素删除。

❌下面是关于暴力解法的初次代码:

class Solution {
public:
    int removeElement(vector<int>& nums,int val){
    int size = nums.size();
    for(int i = 0; i < size; i++){
        if(nums[i] == val){
            for(int j = i + 1; j < size; j++){
                nums[j - 1] = nums[j];
            }
            i--;
            size--;
        }
    }
}
};

这段代码中存在一个错误

for循环没有写返回值,在for循环结束后应该写return size;按题目要求与代码合法性返回。

✅更正后符合力扣格式的代码:

class Solution {
public:
    int removeElement(vector<int>& nums,int val){
    int size = nums.size();
    for(int i = 0; i < size; i++){
        if(nums[i] == val){
            for(int j = i + 1; j < size; j++){
                nums[j - 1] = nums[j];
            }
            i--;
            size--;
        }
    }
    return size;
    }
};

暴力解法的特点是便于理解,一层循环遍历数组,一层循环覆盖前值更新数组。但也应该注意到暴力解法时间复杂度O(n^2)空间复杂度O(1),在一定程度存在优化空间。


方法二 双指针法

所谓双指针法是指通过两个指针——快指针和慢指针,在一个for循环下完成两个for循环完成的任务。其中:

  • 快指针:寻找新数组的元素 ,新数组就是不含有目标元素的数组
  • 慢指针:指向更新新数组下标的位置

❌下面是关于双指针法的初次代码:

class Solution {
public:
    int removeElement(vector<int>& nums,int val){
    int slowIndex = 0;
    for(int fastIndex = 0; fastIndex < nums.size(); fastIndex++){
        if(val != nums[fastIndex]){
            nums[slowIndex] = nums[fastIndex];
        }
    }
    return slowIndex;
}
};

进行运行测试后发现报错:

是因为代码中 慢指针slowIndex 没有实现自增,也就是说慢指针不会后移,只会在0处不断覆盖,导致报错。正确写法有两种,一是直接 nums[slowIndex++]=nums[fastIndex],二是在原代码后加 slowIndex++

✅更正后符合力扣格式的代码:

class Solution {
public:
    int removeElement(vector<int>& nums,int val){
    int slowIndex = 0;
    for(int fastIndex = 0; fastIndex < nums.size(); fastIndex++){
        if(val != nums[fastIndex]){
            nums[slowIndex++] = nums[fastIndex];
        }
    }
    return slowIndex;
}
};