力扣026 删除排序数组中的重复项
题干
给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例
给定数组 nums = [1,1,2],
函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。
你不需要考虑数组中超出新长度后面的元素。
给定 nums = [0,0,1,1,1,2,2,3,3,4],
函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。
你不需要考虑数组中超出新长度后面的元素。
说明
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
解析
这道题一度困扰了我几个小时(属于是菜鸡了),其实一开始很容易被本题当中原地修改数组这个点所吸引注意力,而忽略一个更重要的点——排序数组,排序数组经常和双指针联系到一起,虽然双指针的使用方法每道题有所不同,但遇到排序数组就想到双指针不失为一种思路,只是读完题干后没有马上注意到这一点。
既然想到了要使用双指针,就要确定双指针具体怎么用,指针的初值、指针的遍历方法等都要思考清楚。
观察本题,可以发现一个切入点,输出的数组其实分为了两个部分:
1、不含重复元素的前段
2、全是重复元素的后段
我们能不能从这个角度去定义双指针的工作范围呢?比如指针 i 负责遍历不含重复元素的前段,指针 j 负责遍历全部数组,找出重复元素。
我解本题的思路基本来源于力扣中别人写的题解,其中有一个题解中通过一个简洁的视频解释双指针工作的过程,参考的是【景禹】的视频动画。
另外本题对于额外空间的限制以及原地修改的题干,其实就是要求我们直接修改原数组,不要使用另一个数组解题。
代码
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
# 双指针
i = 0
for j in range(1,len(nums)):
if nums[i] != nums[j]:
i += 1
nums[i], nums[j] = nums[j], nums[i]
return i + 1
一个小地方需要注意,遍历结束后的指针 i 指向最后一个不重复元素,而此时我们需要返回的是一个长度值,因此返回的是 i+1