LeetCode *27. Remove Element (Python Solution)

题目描述

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和值val，在适当位置删除该值的所有实例并返回新长度。

不要为另一个数组分配额外的空间，必须通过使用O（1）额外内存修改输入数组来实现此目的。

元素的顺序可以改变。 你留下的新长度并不重要。

Example 1:

Given nums = [3,2,2,3], val = 3,

Your function should return length = 2, 
with the first two elements of nums being 2.

It doesn't matter what you leave
beyond the returned length.

Example 2:

Given nums = [0,1,2,2,3,0,4,2], val = 2,

Your function should return length = 5, 
with the first five elements of nums 
containing 0, 1, 3, 0, and 4.

Note that the order of those five 
elements can be arbitrary.

It doesn't matter what values are 
set beyond the returned length.

Python Solution

分析： 如果题目中允许运用额外的空间，这道题就是很简单很基础的一道 for 循环语句的练习题。但是需要注意的是题目只允许 O(1) 的空间复杂度，即必须在原来的 list 上进行改动。

这里采取的思想有点像双指针，一开始都是指向 nums[0] 这个元素，然后一个指针作为改变的基准值 idx ，另一个指针去走 for 循环。如果值不等于 val ，就将 nums[i] 赋给 nums[idx] 。这样实现的效果其实就是把所有值不为 val 的元素按照它们原来在列表中的顺序移到了列表的开头。

代码如下：

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        if not nums:
            return None
        idx = 0
        for i in range(len(nums)):
            if nums[i] != val:
                nums[idx] = nums[i]
                idx += 1
        return idx

复杂度分析：

时间复杂度：O(n) ，for 循环从头到尾遍历一次，赋值的时间复杂度为 O(1) 所以没有影响。
空间复杂度：O(1) ，只是改变数组列表，并没有新建操作。