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

代码一思路：

设两个指针cur、i，cur是有效元素下标，初值为0，i是从第一个元素开始的遍历指针；另外用count记录当前相同元素的个数；

i和cur指向的元素相同且count值小于2，说明cur指向的元素出现第二次，为有效元素，cur、i右移，count计数器加1；

i和cur指向的元素相同且count值不小于2，说明cur指向的元素出现已经大于两次，此后均是非有效元素，cur不动，仅i右移；

i和cur指向的元素不同，说明遇到新元素，为有效元素，cur、i右移，count计数器重置为1；

返回值是有效元素个数，即cur+1。

class Solution(object):
    def removeDuplicates(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        n = len(nums)
        if not n:
            return 0
        count = 1
        cur = 0
        i = 1
        while i < n:
            # i和cur指向的元素相同且count值小于2，说明cur指向的元素出现第二次，为有效元素，cur、i右移，count计数器加1；
            if nums[i] == nums[cur] and count < 2:
                cur += 1
                nums[cur] = nums[i]
                count += 1
                i += 1
            # i和cur指向的元素相同且count值不小于2，说明cur指向的元素出现已经大于两次
            elif nums[i] == nums[cur] and count >= 2:
                i += 1
            # i和cur指向的元素不同，说明遇到新元素，为有效元素
            elif nums[i] != nums[cur]:
                cur += 1
                nums[cur] = nums[i]
                count = 1
                i += 1
        return cur + 1

代码二：代码一思路的优化。

因为元素是有序的，可以直接用nums[i]和nums[i-2]进行判断，如果相等，那么说明重复的元素一定超过了两个，当前元素需要跳过。

class Solution(object):
    def removeDuplicates(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        i = 0
        for item in nums:
            if i < 2 or nums[i - 2] != item:
                nums[i] = item
                i += 1
        return i