题目:删除有序数组中的重复项 给你一个 非严格递增排列 的数组 nums,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums中唯一元素的个数。 举例: 输入:nums = [0,0,1,1,1,2,2,3,3,4] 输出:5, nums = [0,1,2,3,4]
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
"""
#方法一:使用额外空间,新建一个列表存储不重复的元素,遍历原列表,每次添加元素判断在新列表中是否存在,存在即不添加到新列表,不存在就添加到新列表,最后新列表里面就是不重复的元素,然后把新列表的值赋给原列表
nums_tmp = []
for i in nums:
if i not in nums_tmp:
nums_tmp.append(i)
for i in range(len(nums_tmp)):
nums[i] = nums_tmp[i]
return len(nums_tmp)
"""
#方法二:不使用额外空间,双指针思路,一个记录最终结果,一个用于判断是否满足要求
curr = 0
back = 1
while back <= len(nums)-1:
#下一个与当前比,如果重复就继续往下遍历
if nums[back] == nums[curr]:
back += 1
continue
#如果不重复,就找到了下一个要填写的正确值
curr += 1
nums[curr] = nums[back]
back += 1
return curr + 1
注:以上懒得改格式了
进阶1:
给你一个有序数组 nums
,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
curr = 1
back = 2
while back <= len(nums) - 1:
#下一个其前面2个比,如果相同继续往下找
if nums[back] == nums[curr] and nums[back] == nums[curr-1]:
back += 1
continue
#找到了,将找到的填写到下个里面
curr += 1
nums[curr] = nums[back]
back += 1
return curr + 1
进阶2:允许每一个元素至多出现K次
class Solution:
def removeDuplicates(self, nums: List[int],k:int) -> int:
curr = k-1
back = k
while back <= len(nums) - 1:
flag = 1
for i in range(k):
flag *= nums[back] == nums[curr-i]
if flag:
back += 1
continue
curr += 1
nums[curr] = nums[back]
back += 1
print(nums)
return curr + 1