27、移除元素--LeetCode

本文介绍了一种在原地删除数组中特定值的高效算法,通过双指针技巧实现,确保了O(1)的空间复杂度。提供了两种不同的实现思路及对应的Python和Java代码示例。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目描述

给定一个数组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中的前五个元素为01304。
注意这五个元素可以为任意顺序。
你不需要考虑数组中超出新长度后面的元素。

说明:
为什么返回数值是整数,但输出的答案是数组呢?

请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝
int len = removeElement(nums, val);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

注意:
1、当遍历元素的时候逐个比较元素的时候,如果出现和给定的值相同的元素的时候是不能删除的,因为使用删除之后,列表的数目就会减小,之前计算的列表的长度就会出现变化,当再次运行的时候会出现溢出的现象(index out of range),例如:

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        if nums is None: return 0
        n = len(nums)
        i = 0
        for j in range(0,n):
            if nums[j] == val:
                # 不能使用删除,删除之后n的值会变化,出现溢出错误!
                del nums[j]
                i += 1
        return n-i

2、题目说明是原地修改,说明需要使用指针,对待元素的遍历,删除应该使用覆盖的方法。
思路1:下一个元素覆盖前一个元素,使用双指针
其实就是使用题目提示的一个信息,元素的顺序可以任意!
1、如果当前的传入的nums的值为空值,那么就返回0.
2、定义一个快指针一个慢指针j和i,慢指针记录覆盖的次数,就是最后数组的长度。
3、快指针覆盖慢指针的值:遍历出入来的数组,如果数组中当前值不等于传入的值val,那么就将快指针对应nums中的值覆盖掉慢指针对应的nums的值,同时慢指针递增1,否者快指针就迅速遍历取得下一个值,继续上面的步骤。
4、循环3,至结束,返回慢指针记录的值就是数组的长度。
时间复杂度O(n),假设数组的长度为n
空间复杂度O(1),仅仅使用两个指针
python3实现代码逻辑如下:

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        if nums is None: return 0
        n = len(nums)
        i = 0
        for j in range(0,n):
            if nums[j] != val :
                nums[i] = nums[j]
                i += 1
        return i

思路2:使用末尾的值向前覆盖,使用双指针
1、当传入的nums的值的长度为空值的时候,返回0
2、定义一个指向开头的指针并初始化j=0和一个指向数组末尾的指针n(n的值其实就是传入的数组的长度,先将n计算出来)
3、遍历数组nums,如果数组的第一个元素等于给定的val的时候,就将最后的元素覆盖到前面的值,控制最后的元素递减1,头指针不动,否则头指针向后移动一个元素,继续进行比较。
4、循环步骤3,直至结束
5、返回的租后长度n就是新的数组的长度。
时间复杂度:O(n),数组的长度最大为n。
空间复杂度:O(1),使用两个指针。
Java实现代码如下:

class Solution {
    public int removeElement(int[] nums, int val) {
        int n = nums.length;
        for (int j = 0; j < n;) {
            if (nums[j] == val) {
                nums[j] = nums[n - 1];
                n--;
            } else {
                j++;
            }
        }
        return n;   
    }
}

注意:
当使用Python的时候,由于使用的for循环是一个迭代的对象,我们是不能像java一样可以直接在for循环里面抽出j++来进行控制的,因为for就是对对象进行迭代,不存在j++这种东西,所以我们需要换一个循环while来达到我们的目的:
使用Python3实现以上的算法逻辑如下:

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        if nums is None: return 0
        n = len(nums)
        j = 0
        while j >=0 and j <n:
            if nums[j] == val :
                nums[j] = nums[n-1]
                n -= 1
            else:
                j += 1
        return n
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值