【python算法与数据结构入门日志】2：数组-移动元素

本文链接：https://blog.csdn.net/hotpants/article/details/128504042

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

来源：https://leetcode.cn/leetbook/read/all-about-array/x9p1iv/

示例：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,4,0,3]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

方法一：单指针正序查找

正向查找时注意删除指令执行后（删除某项后）指针应该停留在当前位置的问题。

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

方法二：从后往前遍历

逆向查找不用担心删除项后数组整体前移的问题。

# python3 method1 从后往前遍历
def removeElement(self, nums, val):
    n = len(nums)
    for i in range(n-1,-1,-1):
        if nums[i] == val:
            del nums[i]
    return len(nums)
# 作者：WEI
# 链接：https://leetcode.cn/leetbook/read/all-about-array/x9p1iv/?discussion=SMcans
# 来源：力扣（LeetCode）
# 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

错误示例：

def removeElement(self, nums, val):
    for i in nums:
        if i == val:
            nums.remove(i)
    return len(nums)

输入：[1,2,3,3,3,3,4,5]
输出：6
问题：删除掉的3为：[1,2,3 ,3,3 ,3,4,5]；问题出在语句for i in nums上，因为for循环块中存在remove，在循环的过程中nums是在不断变化的；而for i in nums是由指针指向nums数组中的序号驱动的，也就是说：当i指向第一个3时（nums[2])，执行remove语句删除了第一个3，同时数组自动进行向前填补空缺，这时填补了原来nums[3]的就是原数组的第二个3；而这时在进行下一波循环时，for指向的nums[3]实际上是原数组的第三个3，这样就导致第二个3被跳过，没有被监测到；第四个3同理也被跳过了，因此最终输出的数组是：[1,2,3,3,4,5]，长度是6.