给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/remove-element
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法一:暴力遍历
暴力解法的时间复杂度是O(n^2),不推荐
/*
可能会想直接删掉多余的元素,不就可以了,但是数组的元素在内存地址中是连续的,不能单独删除数组中的某个元素,只能覆盖。
*/
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int size = nums.size();
//外层for循环遍历数组元素
//内层for循环更新数组
for(int i = 0; i < size; ++i) {
if(nums[i] == val) {
//找到了,就把num[i]之后的元素都向前挪动一位
for(int j = i + 1; j < size; ++j) {
nums[j - 1] = nums[j];
}
--i; //因为下标i以后的数值都向前移动了一位,所以i也向前移动一位
--size; //此时数组的大小需要-1
}
}
return size;
}
};
方法二:双指针
/*
· 如果右指针指向的元素不等于val,他一定是我们需要输出的元素,所以将右指针指向的元素复制到左指针位置,同时左右指针均向右移
· 如果右指针指向的元素等于val,它不能在输出数组里,此时左指针不动,右指针右移一位
*/
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int left = 0;
int size = nums.size();
for(int right = 0; right < size; ++right) {
if(nums[right] != val) {
nums[left++] = nums[right];
}
}
return left;
}
};
//时间复杂度:O(n),其中 n 为序列的长度。我们只需要遍历该序列至多两次。
//空间复杂度:O(1)。我们只需要常数的空间保存若干变量