题目概要
- 给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
题解概述
本人本次解题用时:25min
核心思想:双指针移动,最终将数组nums分为左右两部分,左边是要的,右边是不要的。令左指针为left,右指针为right。
- 法一:双指针。left下一个将要赋值的位置(起点为0处);right指向当前将要处理的元素(起点为0处)。当right所指不等于val时,将right值复制给left值,然后left和right同时右移;当right所指的值等于val时,left不动,right继续右移。最终,[0,left)中的元素都不等于val;终止条件为right遍历完成;left值就是输出数组的长度。
- 法二:双指针优化(从双指针分别遍历数组一次——>双指针合起来遍历数组一次)。left(起点为0处);right(起点为n处)。当left所指的值等于val时,将right所指的值复制给left值,right左移。若复制过来的值恰好等于val时,则继续复制,right继续左移,直到left所指不等于val。最终,left指针与right指针重合。
代码记录
//法一(C++版)
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int n = nums.size();
int left = 0;
for(int right = 0; right<n; right++){
if(nums[right] != val){
nums[left] = nums[right];
left++;
}
}
return left;
}
};
//法二(C++版)
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int left = 0,right = nums.size();
while(left < right){
if(nums[left] == val){
nums[left] = nums[right-1];
right--;
}else{
left++;
}
}
return left;
}
};