给定一个增序排列数组 nums ,你需要在 原地 删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例 1:
输入:nums = [1,1,1,2,2,3]
输出:5, nums = [1,1,2,2,3]
解释:函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。 你不需要考虑数组中超出新长度后面的元素。
思路:一开始的想法是创建字典,通过字典的计数来判断同样元素有几个,大于2的就让他等于2,以保证数量符合要求,结果是很正常的失败了(
换个思路,先观察示例1的数组nums,发现有3个1,这时候很容易找到一个规律,如果某个数的数量大于2了,那么此时的nums[i]和nums[i-2]的值一定相等,利用这个规律可以搞定。
代码:(这部分其实有错误,因为是我最开始写的)
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
i = 2
length = len(nums)
while i < length:
if nums[i] == nums[i - 2]:
nums.pop(i)
else:
i = i + 1
return len(nums)
这样提交之后报出了溢出的错误,原因在于length = len(nums) 这句,在下面进行比较的时候,pop出去的值已经不存在nums中,因此nums的长度是变化的,所以去掉这句话,下面改成while i < len(nums) 就不会出现溢出的问题了。
题目来自LeetCode第80题