python力扣刷题记录——27. 移除元素

题目:

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

使用双指针法
方法一:
执行用时: 36 ms
内存消耗: 13.5 MB
当nums[i] 与给定的值相等时,递增 i 以跳过该元素。只要valnums[i] != val,我们就复制 nums[i] 到 nums[j] 并同时递增两个索引。重复这一过程,直到 i 到达数组的末尾,该数组的新长度为 j。

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

方法二:
执行用时: 36 ms
内存消耗: 13.5 MB
当我们遇到 nums[i] = valnums[i]=val 时,我们可以将当前元素与最后一个元素进行交换,并释放最后一个元素。这实际上使数组的大小减少了 1。
请注意,被交换的最后一个元素可能是您想要移除的值。但是不要担心,在下一次迭代中,我们仍然会检查这个元素。

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

猜你喜欢

转载自blog.csdn.net/weixin_45455015/article/details/110848928