给你一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路1:遍历数组,找到所有等于val的值,后面的元素全部前移
时间复杂度:最坏情况下,O(N^2)
思路2:重新开辟一个数组tmp,遍历原数组,把不是val的数全部放到新数组中tmp,再把tmp数组拷贝到原数组中
时间复杂度:O(N)
空间复杂度为:O(N)
思路3:使用双指针(下标),src、dst,src去寻找nums中不等于val的值,放到dst指向的位置,然后src++,dst++
也就是把不是val的数字集中到数组前半部分,当nums[src]的值不等于val时,将nums[src]的值赋值给nums[dst],同时src++,dst++,当nums[src]等于val时,dst不变,src++。
空间复杂度为O(1)
假设nums = [0,1,2,2,3,0,4,2], val = 2
代码实现:
int removeElement(int* nums, int numsSize, int val){
int src = 0;
int dst = 0;
while (src < numsSize)
{
if (nums[src] != val)
{
nums[dst] = nums[src];
src++;
dst++;
}
else
{
src++;
}
}
return dst;
}