给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
本题采用快慢指针法
思想:
设置快指针循环遍历整个数组,而慢指针的作用是呆在val值所在的地方,用fast指向的不为val的值来代替它。
当没有遇到val值的时候,fast指针和slow指针是同时往前移动的,即指向同一个值。
一旦在数组中遇到了val值,slow指针就会停下来。直到当fast指针所指的值不为val值,就可以把fast里面值放到slow里,并且slow往后移动。此后fast和slow会一起往后移动,并且fast只有碰到不为val的值,才会把它放到slow指针所指的位置。
直到fast后面遍历完整个数组之后,slow指针前就是这个数组全部不为val的值。
int removeElement(vector<int>& nums, int val) {
//双指针解法
int slow = 0;
for (int fast = 0; fast < nums.size(); fast++) {
if (nums[fast] != val) {
nums[slow] = nums[fast];
slow++;
}
}
return slow;
}