[简单]LeetCode 27.移除元素

题目

题解

第一想法

这道题目有个知识点：在函数里修改输入数组对于调用者是可见的。这个点我印象非常深刻。去年在用c刷题的时候就着重理解了一波这个问题。
题目给的解释已经非常清晰明了了，这里不做过多赘述。

这道题第一反应就是可以将==val的元素和末尾进行交换。但是如果测试用例的末尾元素也是val那不就翻车了奥？！所以这里需要使用到while，如果nums[i]==val并且i<numSize的时候，我们继续进行末尾元素交换。但是在上一回操作的时候numSize已经自减了，所以第二回其实是在和倒数第二个元素进行交换。那如果交换一次之后不再等于val那么就进行下一个i判断。

正确题解

int removeElement(int* nums, int numsSize, int val){
    for(int i=0;i<numsSize;i++){
        if(nums[i]==val){
            while(i<numsSize && nums[i]==val){
                int tmp = nums[i];
                nums[i] = nums[numsSize-1];
                nums[numsSize-1] = tmp;
                numsSize--;
            }
           
        }
    }
    return numsSize;
}

或者另一种思路，！=val的时候，直接保存替换。

int removeElement(int* nums, int numsSize, int val){
    int nowid=0;
    for(int i=0;i<numsSize;i++){
        if(nums[i]!=val){
            nums[nowid++]=nums[i];
        }
    }
    return nowid;
}