本文共 818 字,大约阅读时间需要 2 分钟。
给定一个增序排列数组 nums ,你需要在 原地 删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
设置两个指针,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.
最后返回列表的长度。
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)
–
转载地址:http://idagz.baihongyu.com/