题目叙述:给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
题目中要求必须仅使用O(1)额外空间并原地修改输入数组。如果没有这个要求的话,我们可以考虑新建立一个数组,然后遍历原来的数组,对于不是Val的数字可以放在新数组里。
然而这道题目我们可以这样做考虑:使用两个指针,prev和cru,使他们都从数组的头开始遍历,如果prev指向的数字不是目标元素val,则将prev指向的元素赋值给cru,如果是val的话,只让prev向后走并判断下一个是否是目标元素,如果是的话接着往下走直到遇到不是目标元素的数字,并将它赋值给cru。
最后返回的cru就是数组的长度
int removeElement(int* nums, int numsSize, int val)
{
int cru = 0;
int prev = 0;
while(prev<numsSize)
{
if (nums[prev] != val)
{
nums[cru] = nums[prev];
cru++;
prev++;
}
else
{
prev++;
}
}
return cru;
}