題目:
"""
給你一個有序數組 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)