力扣 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 <= 1000 <= nums[i] <= 500 <= 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;
}
};