leetcode学习笔记一：移除元素（python）

最新推荐文章于 2022-11-04 23:26:48 发布

爱琢磨先森

最新推荐文章于 2022-11-04 23:26:48 发布

阅读量201

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode学习 python

本文链接：https://blog.youkuaiyun.com/m0_37952909/article/details/89342018

本文探讨如何在不使用额外空间的情况下，高效地从数组中移除特定值。通过对比不同循环方式，如倒序循环、正循环及while循环，结合双指针法，实现数组元素的原地修改与移除，同时保持O(1)的空间复杂度。

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

题目描述：

给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

例：

给定 nums = [3,2,2,3], val = 3,

函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。

你不需要考虑数组中超出新长度后面的元素。

体会：

1.学会使用倒序循环

有效避免了正循环时数组越界和跳过元素的问题

#倒序循环
def removeElement(nums,val):
    j = len(nums)
    for i in range(j - 1, -1, -1):
        if nums[i] == val:
            nums.pop(i)
    return len(nums)

#正循环
def removeElement(nums,val):
    j = len(nums)
    for i in range(j):
        if nums[i] == val:
            nums.pop(i) #每删除一个元素，会改变数组的大小，i在不断增加，会跳过元素且会出现indexvalue
    return len(nums)

#可使用while循环

def removeElement(nums,val):
    i=0
    while i<len(nums):
        if nums[i]==val:
            nums.pop(i)
        else:
            i=i+1
    return len(nums)

2.类似java的双指针

从头到尾遍历，正循环，不存在计数，并往前覆盖

def removeElement(nums,val):
    j = 0
    for i in range(len(nums)):
        if nums[i] != val:
            nums[j] = nums[i]
            j += 1
    return j