Description
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使每个元素 最多出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
输入:nums = [1,1,1,2,2,3]
输出:5, nums = [1,1,2,2,3]
解释:函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。 不需要考虑数组中超出新长度后面的元素。
思路
这个思路还有点难想。考虑一个边界情况,当index<2
的时候:这个时候不会出现长度超标,可以直接全放进去。
index超过2,就有可能出现超长的情况。这时候注意到超长成员的特征:e==nums[i-2]
。考虑到此时index>=2
(index是从0开始的),所以不会出现越界访问。如果出现超长,那么这里就不再录入。
代码
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
i = 0
for e in nums:
if i < 2 or e != nums[i - 2]:
nums[i] = e
i += 1
return i
考虑到题目要求原地算法,所以把符合要求的成员丢前面去,用一个慢指针存储。
感觉还是说不太清楚。。。如果没看懂的话可以看看官方题解加深理解。