数据结构之移除元素

（leetcode刷题）

移除元素

  题目： 给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1)额外空间的条件下完成。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
  示例：nums = [3,2,2,3], val = 3，返回值为2，且nums前两个元素为2；nums = [0,1,2,2,3,0,4,2], val = 2，返回之为6，且nums前5个元素为[0, 1, 3, 0, 4]（任意顺序）。后面的元素不用考虑。

方法一

• 我的想法：遍历所有元素，删除相同元素。（leetcode多提交几遍会变快）

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

方法二：双指针法

• 移除元素这题与删除重复项很相似，都可以使用双指针法。不过运行下来和上面的速度差不多。

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