题目
给一个数组 nums 和一个值 val,原地移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,必须仅使用 O(1) 额外空间并原地修改输入数组。
元素的顺序可以改变,不需要考虑数组中超出新长度后面的元素。
题解示例
输入:nums = [3,1,4,3], val = 3
输出:2, nums = [1,4]
解释:函数应该返回新的长度 2。例如,nums = [1,4,3,3] 或 nums = [1,4,0,0],也会被视作正确答案。
题目要求原地删除等于 val 的元素,可以采用双指针的方法,在原数组上重新赋值得到符合要求的新数组。
左指针 left 指向下一个赋值的位置,右指针 right 指向当前处理的元素。最坏情况下,遍历数组两次。
优化
上述方法中,若某一元素不满足条件,则后面满足条件的元素将不断向前赋值。为了避免满足条件的元素的重复赋值操作,可以设置双指针从左右两边向中间移动遍历序列。
左指针 left 指向当前处理的元素,右指针 right 指向未处理的最后一个元素。左指针从左向右移动,若指向元素等于 val ,则将右指针指向元素复制到左指针的位置,然后右指针左移,继续判断左指针指向元素是否等于 val ;若指向元素不等于 val ,则左指针右移,继续判断下一个元素。
更多题解
代码- 若指向元素等于 val ,则右指针左移
- 若指向元素不等于 val ,则左指针右移
class Solution {
public:
int removeElement(vector& nums, int val) {
//定义双指针
int left = 0, right = nums.size()-1;
//当 left > right 时遍历完数组
while (left <= right) {
if (nums[left] == val) {
//将右指针指向元素复制到左指针指向位置,避免要保留元素的重复赋值操作
nums[left] = nums[right];
right--;
} else {
left++;
}
}
//返回数组长度
return left;
}
};



