27题

第27题
因为Python循环内部隐藏了记录当前迭代器的状态,而pop方法删除了当前元素后,迭代器自动指到下一个元素,原先的list在内存中的位置并未改变。
比如[0,1,2,2,3,0,4,2],val=2,第一个2被弹出后,第二个2成为了nums[2],而此时的i将变成3,导致没法弹出第二个2.

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
    '此为错误解法'
        i=0
        while (i<len(nums)):
            if nums[i] == val:
                nums.pop(i)
            i+=1
        return len(nums)

因此加上else语句,一开始的解法

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        i=0
        while (i<len(nums)):
            if nums[i] == val:
                nums.pop(i)
            else:
                i+=1
        return len(nums)

改进: 反向迭代,可以避免list变化导致漏掉元素的情况

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        for i in range(len(nums)-1,-1,-1):
            if(nums[i] == val):
                nums.pop(i)
        return len(nums)

或者使用非迭代手段(不推荐)

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        while val in nums:
            nums.remove(val)
        return len(nums)

双指针法

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        '双指针法'
        i = 0
        for j in range(len(nums)):
        # j为快指针,i为慢指针
            if nums[j] != val:
                nums[i] = nums[j]
                i += 1
        return i

四种写法时长相差不大

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值