26.删除排序数组中的重复项
题目
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不用使用额外的数组空间,你必须在原地修改输入数组并在使用O(1)额外空间的条件下完成。
示例1:
给定数组nums=[1,1,2],
函数应该返回新的长度2,并且原数组nums的前两个元素被修改为1,2。
你不需要考虑数组中超出新长度后面的元素。
示例2:
给定 nums = [0,0,1,1,1,2,2,3,3,4],
函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。
你不需要考虑数组中超出新长度后面的元素。
一开始我的思路是:
前后覆盖,遇到num[slower]==num[faster] faster +=1,然后再覆盖
然后写出的代码是这样
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
#前后覆盖,遇到num[slower]==num[faster] faster +=1,然后再覆盖
if (len(nums)<=0): return 0
slower,faster=0,0
while(faster<len(nums)):
nums[slower]=nums[faster]
if nums[slower]==nums[faster]:
faster+=1
slower+=1
faster+=1
nums=nums[:slower]
# print(slower,faster)
return len(nums)
并侥幸通过了示例1的测试,但是在示例2的测试中失败了。我想了一下确实,我的思路存在漏洞,就是会漏掉最后一个元素,因为出现相同数字的时候才会去隔一个数进行覆盖。而正确的思路应该是这样:
慢指针指向待写入元素的位置,快指针遍历数组,当快指针指向的元素和慢指针不同时,说明相同元素已经遍历结束,此时将慢指针后移,将快指针的元素写入慢指针位置,保留一个元素。即它的思路是找相同元素,当找不到相同元素的时候,就让慢指针后移动,将快指针指向元素写入慢指针的位置,然后以新的慢指针指向的元素作为新的遍历对象继续寻找与其相同的元素。
代码如下:
class Solution:
def removeDuplication(self,nums:List[int])->int:
if not nums:return 0
slower,faster=0,0
while(faster<len(nums)):
if nums[slower]!=nums[faster]:
slower+=1
nums[slower]=nums[faster]
faster+=1
return slower+1
测试用例
def removeDuplicates(nums):
if not nums: return 0
slower,faster=0,0
while(faster<len(nums)):
if nums[slower]!=nums[faster]:
slower+=1
nums[slower]=nums[faster]
faster+=1
return slower+1,nums
nums= [0,0,1,1,1,2,2,3,3,4]
index,result=removeDuplicates(nums)
print(result[:index])