LeetCode 80删除有序数组中的重复项II Python3题解

# 删除有序数组中的重复项 II
"""
给你一个有序数组nums,请你原地删除重复出现的元素，使得出现次数超过两次的元素只出现两次，返回删除后数组的新长度。
不要使用额外的数组空间，你必须在原地修改输入数组并在使用O(1)额外空间的条件下完成。
"""

"""
    解题思路其实就是一个双指针思想  有一个不完善版本Solution1 有一个完善版本Solution
"""


class Solution1:
    def removeDuplicates(self, nums):  # 阉割版的方法 保证不了最后一个
        count = 1  # 出现的次数
        j = 0  # 满足要求的数组索引
        for i in range(len(nums) - 1):
            if nums[i] == nums[i + 1]:
                count += 1
            else:
                d = min(count, 2)
                for k in range(d):
                    nums[j] = nums[i]
                    j += 1
                count = 1
        return j


print(Solution1().removeDuplicates([0, 0, 1, 1, 1, 1, 2, 3, 3]))


class Solution:
    def removeDuplicates(self, nums):  # 可以保证最后一个
        count = 1  # 出现的次数 count就是这个数据出现了几次 主要就是防止出现了一次
        j = 0  # 满足要求的数组索引
        for i in range(len(nums)):  # [0, 0, 1, 1, 1, 1, 2, 3, 3]
            if i < len(nums) - 1 and nums[i] == nums[i + 1]:
                # 要注意i不能到最后一个 要不然i+1报错
                # 当前位置的值与下一个位置的值是一样的就把count增加
                count += 1
            else:  # 当当前值与下一个不一样 就更新 包括最后一个元素也是可以这样处理
                d = min(count, 2)  # 这个地方主要就是为了防止出现一个的元素
                for k in range(d):  # 一般就是更新两次或一次
                    nums[j] = nums[i]
                    # nums[0]=nums[1] nums[1]=nums[1]  
                    # nums[2]=nums[5] nums[3]=nums[5]
                    # nums[4]=nums[6]
                    # nums[5]=nums[8] nums[6]=nums[8]
                    # 这个是样例的更新过程可以看一下
                    j += 1
                count = 1  # 当你更新完之后 下次count又要设为1了 重新统计
        return j   # 函数应该返回新长度就是修改了几个位置 自然是7个 也就是j


print(Solution().removeDuplicates( [0, 0, 1, 1, 1, 1, 2, 3, 3]))