1. 程式人生 > 其它 >leetcode-移除有序陣列重複元素[26]

leetcode-移除有序陣列重複元素[26]

題目描述

給你一個 升序排列 的陣列 nums ,請你 原地 刪除重複出現的元素,使每個元素 只出現一次 ,返回刪除後陣列的新長度。元素的 相對順序 應該保持 一致 。

示例

 輸入;nums=[1,1,2]
 輸出2,nums=[1,2,_]

雙指標

使用fast指標在前面探路,若遇到重複元素,則跳過,否則(遇到有效元素),slow+1,並把fast指向的元素複製到slow所指的元素。
注:[0:slow] 有效陣列,保留不重複元素,slow+1為新陣列長度。
注意本題和leetcode-移除元素[26]的區別,slow要先加1,然後賦值

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        n = len(nums)
        fast = slow=0

        while fast<n:
            # 遇到有效元素
            if nums[fast]!=nums[slow]:
				# slow先移動,後賦值
                slow+=1
                nums[slow] = nums[fast]
            fast+=1
        # return nums[:slow+1]
        return slow+1