1.题目详情
给定一个增序排列数组 nums ,你需要在 原地 删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
2.解题思路
设置两个指针,i,count。i负责遍历列表,count负责记录每个数字的次数,count初始为1。
i 在遍历时
-
如果nums[i] == nums[i-1],也就是与前一项数字相等,那么count计数加一。而由题意知,count只能小于等于2,因而当count大于2时,删除nums[i],并将索引减一。
-
如果nums[i] != nums[i-1],说明是新的数字,那么count更新为新的1.
最后返回列表的长度。
3.代码实现
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
i,count = 1,1 # i遍历,count计数
while i < len(nums):
if nums[i] == nums[i-1]: # 若与前一项数字相等,计数器+1
count += 1
if count > 2: # 若计数器大于2,将移除一个重复的数,并且索引往前移一位
nums.remove(nums[i])
i -= 1
elif nums[i] != nums[i-1]: # 若与前一项数字不等,计数器更新为新的1
count = 1
i += 1
return len(nums)
4.知识点
–