leetcode80. 删除排序数组中的重复项 II

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素最多出现两次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

示例 1:

给定 nums = [1,1,1,2,2,3],

函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。

你不需要考虑数组中超出新长度后面的元素。
示例 2:

给定 nums = [0,0,1,1,1,1,2,3,3],

函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3 。

你不需要考虑数组中超出新长度后面的元素。

 

用快慢指针可以写出通解算法，即“删除排序数组中的重复项，使得相同数字最多出现 k 次”

 

• 初始化快慢指针 slow ， fast ，全部指向索引为 0 的元素。
• fast 每次移动一格
• 慢指针选择性移动，即只有写入数据之后才移动。是否写入数据取决于 slow - 2 对应的数字和 fast 对应的数字是否一致。
• 如果一致，我们不应该写。 否则我们就得到了三个相同的数字，不符合题意
• 如果不一致，我们需要将 fast 指针的数据写入到 slow 指针。
• 重复这个过程，直到 fast 走到头，说明我们已无数字可写。

参考自https://github.com/azl397985856/leetcode/blob/master/problems/80.remove-duplicates-from-sorted-array-ii.md

图解（红色的两个数字，表示我们需要比较的两个数字）：

 

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        i = 0
        K = 2
        for num in nums:
            if i < K or num != nums[i-K]:
                nums[i] = num
                i += 1
        return i