Leetcode 27. 移除元素 | 数组OJ

题目链接

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

思路分析

法一：看到这样的题，一开始的思路是遍历数组，当数值和value一样的时候，将之后的数组向前移动，在这样情况下的时间复杂度是O（N^2)。
法二：

1. 设定 after 和 pre 两个下标，初始化下标的值都是0，当pre下标比numsSize小的时候，连续遍历数组。
2. 当pre下标对应的数组的值和val一样的时候， pre++；
3. 当pre下标对应的数组的值和val不一样的时候，把 pre下标对应的数组的值赋值给 after下标对应的值，然后pre++，after++；
4. 因为数组的长度等于数组最后一个元素下标+1，并且对于最后一个不同于val的值，after继续++了，所以最后返回after的值就是数组的长度。

题解

int removeElement(int* nums, int numsSize, int val){
    int after = 0, pre = 0;
    while(pre < numsSize)
    {
        if(nums[pre] == val)
        {
            pre++;
        }
        else
        {
            nums[after] = nums[pre];
            after++;
            pre++;
        }
    }
    return after;
}