给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
【27.移除数组】示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
方法一
将需要移除的数据与数组的最后一位交换位置,然后删除最后一位,再次判断当前位置的数据,避免交换后的数也是需要移除的数据。
class Solution {
public:
int removeElement(std::vector& nums, int val) {
int i = 0;
int numOf = nums.size();
while (i < numOf)
{
(nums[i] == val) ? (nums[i] = nums[numOf - 1], numOf -= 1) : (i++);
}
returnnumOf;
}
};
方法二
双指针法:i为慢指针,j为快指针,如果某一位的数不是需要移除的数,i和j同时递增,如果某一位的数是需要移除的数,只递增j。
int removeElement2(std::vector& nums, int val)
{
int i = 0;
for (int j = 0;
j < nums.size();
j++) {
if (nums[j] != val) {
nums[i] = nums[j];
i++;
}
}
return i;
}