题目:
"""
给你一个有序数组 nums ,请你原地删除重复出现的元素,使每个元素只出现一次,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组 并在使用 O(1) 额外空间的条件下完成。
"""
法一:在原数组里面删除重复元素
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
if not nums:
return 0
i = 0
while i < len(nums):
if len(nums)==1:
break
if nums[i]==nums[i-1]:#在原数组里面删除重复元素
del nums[i]
else:
i += 1
return len(nums)
法二:双指针
fast指针用来扫描,slow指针用来指向答案位置
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
if not nums:
return 0
n = len(nums)
fast = slow = 1 #fast指针用来扫描,slow指针用来指向答案位置
while fast < n:
if nums[fast] != nums[fast - 1]: #找到不重复项时,将fast位置复制给slow
nums[slow] = nums[fast]
slow += 1
fast += 1
return slow
nums = [0,0,1,1,2]
S = Solution()
result = S.removeDuplicates(nums)
print(result)