LeetCode *26. Remove Duplicates from Sorted Array (Python Solution)

题目描述

Given a sorted array nums, remove the duplicates in-place such that each element appear only once 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.

给定排序的数组nums，就地删除重复项，使每个元素只出现一次并返回新的长度。

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

Example 1:

Given nums = [1,1,2],

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

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

Example 2:

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

Your function should return length = 5, 
with the first five elements of nums 
being modified to 0, 1, 2, 3, and 4 
respectively.

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

Python Solution

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

这里采取的思想有点像双指针，一开始都是指向 nums[0] 这个元素，然后一个指针作为改变的基准值 idx ，另一个指针 i 去从 1 开始走 for 循环（因为是排序列表，所以第一个值一定是最小的不会改变）。
如果值发生了改变不等于 nums[idx] ，就先将 idx + 1，再将 nums[i] 赋给 nums[idx] 。这样实现的效果其实就是 nums[idx] 一直是现在见到的最小值，一旦 nums[i] 和这个值不一样，即有了更大的值。

代码如下：

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

复杂度分析：

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