27. 移除元素 - 力扣(LeetCode)
题目描述
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
样例输入
示例 1:
输入:nums = [3,2,2,3], val = 3 输出:2, nums = [2,2] 解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
示例 2:
输入:nums = [0,1,2,2,3,0,4,2], val = 2
输出:5, nums = [0,1,4,0,3]
题解
这道题使用双指针进行求解时,其思路其实刚好相反。
正常来讲,我们使用暴力解法时,需要判断当前元素与val是否相等,如果相等,我们就将该元素进行移除,移除的办法是通过将该元素后的所有元素均向前移动一个位置,将该重复的元素进行覆盖即可。
而使用双指针解法,本质上遍历nums,然后取出所有与val不相等的值,将其放入一个新的数组中,这样就变相的移除了数组中的所有值为val的元素。
因此我们首先需要一个遍历原数组nums的指针j,其作用是判断nums[i]是否与val相等,如果不相等就将该元素放到“新数组”中,因此在这里指针i起到的作用其实代表“新数组”中元素位置的下标,只不过在这里的“新数组”并不是我们开辟的实体数组,而是直接使用原数组进行构造
总结:
- 指针j遍历原数组,取出与val不相等的元素,将其放入到指针i指向的元素下标中
代码
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int i=0;
for(int j=0;j<nums.size();j++)
{
if(nums[j]!=val)
nums[i++]=nums[j];
}
return i;
}
};