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

题目描述

Given a sorted array nums, remove the duplicates in-place such that duplicates appeared at most twice 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,1,2,2,3],

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

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

Example 2:

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

Your function should return l
ength = 7, with the first seven elements
of nums being modified to 
0, 0, 1, 1, 2, 3 and 3 respectively.

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

Python Solution

分析： 这道题目是在 LeetCode 27 LeetCode 26 的基础上改动的一道题，将出现一次改为出现两次，只需要在原来的基础上加一个 count 用来计数即可。需要注意的是题目只允许 O(1) 的空间复杂度，即必须在原来的 list 上进行改动。

这里采取的思想有点像双指针，一开始都是指向 nums[0] 这个元素，然后一个指针作为改变的基准值 idx ，另一个指针 i 去从 1 开始走 for 循环（因为是排序列表，所以第一个值一定是最小的不会改变）。

如果值发生了改变不等于 nums[idx] ，就先将 idx + 1，再将 nums[i] 赋给 nums[idx] 并且将 count 清零 。

如果没有改变则判断这个值出现次数是不是达到了两次，如果没到则交换移动到 nums[idx] ，如果到了两次则忽略不管。

这样实现的效果其实就是 nums[idx] 一直是现在见到的最小值，一旦 nums[i] 和这个值不一样，即有了更大的值。

代码如下：

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

复杂度分析：

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