[LeetCode] 27. 移除元素

移除元素

题目描述
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。

示例2:
给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。

使用语言: C语言
思路解析: 空间复杂度为O(1),所以不能开辟新空间,只能在原数组上操作,可以采用覆盖的方式移除,给两个下标src和dst都指向下标为0的位置,如果nums[src]不等于val,将nums[src]赋值给nums[dst],++src,++val,如果nums[src]等于val,++src到下一个不是val的数,继续如上情况一赋值,直到src=numsSize结束
代码实现

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

猜你喜欢

转载自blog.csdn.net/weixin_50886514/article/details/113145701