(Leetcode) 移除元素 - Python实现

题目：移除元素

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

Given an array nums and a value val, remove all instances of that value in-place and return the new length.Do not allocate extra space for another array, you must do this by modifying the input array in-place with O(1) extra memory.The order of elements can be changed. It doesn't matter what you leave beyond the new length.

示例:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。

说明:
为什么返回数值是整数，但输出的答案是数组呢?
请注意，输入数组是以“引用”方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

---------------------------------------------------------------------------

思路：这个题目容易出错的地方是，直接用for循环对list遍历删除元素，会导致下标index冲突。因为在循环中删除元素，list的长度一直在变化。

解法1: 用 while 循环 + python的 in 方法 + remove() 函数，代替常规的for循环遍历。

class Solution(object):
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """

        while val in nums:
            nums.remove(val)
        return len(nums)

解法2: 先遍历一遍nums，记录下所有等值val的小标。再通过对marks的遍历，逐个删除元素。

remove() 函数用于移除列表中某个值的第一个匹配项。

class Solution(object):
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """

        marks = []
        for num in nums:
            if num == val:
                marks.append(num)

        for mark in marks:
            nums.remove(mark)
        return len(nums)

解法3：通过while循环 + python自带的pop()方法实现直接删除。

class Solution(object):
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """

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

解法4：通过python自带的count() 和remove()函数实现。

class Solution(object):
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """
        coun = nums.count(val)
        for index in range(coun):
            nums.remove(val)

参考：

https://www.runoob.com/python/att-list-remove.html

https://www.runoob.com/python/att-list-pop.html

https://blog.csdn.net/qiubingcsdn/article/details/82144173

https://blog.csdn.net/qq_34364995/article/details/80274162