python 移除元素

118 阅读1分钟

题目:

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

不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

def removeElement(nums,val):
    new_nums = []
    for num in nums:
        if num == val:
        # 将等于val的值放在一个新列表里
            new_nums.append(num)
            
    for i in new_nums:
    # 遍历旧列表,如果是新列表的值,则移除
    # 直接遍历旧列表,如果旧列表里存在重复的元素,会移除不干净
        nums.remove(i)
    return len(nums)