leetcode題目26. 删除排序數組中的重複項

題目描述

給定一個排序數組，你需要在 原地 删除重複出現的元素，使得每個元素隻出現一次，傳回移除後數組的新長度。

不要使用額外的數組空間，你必須在 原地 修改輸入數組 并在使用 O(1) 額外空間的條件下完成。

示例

給定數組 nums = [1,1,2],

函數應該傳回新的長度 2, 并且原數組 nums 的前兩個元素被修改為 1, 2。

你不需要考慮數組中超出新長度後面的元素。

python代碼

版本1

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        nums.sort()
        n = len(nums)
        if n == 0:
            return 0
        ans = []
        count = 0
        for i in range(n):
            if nums[i] not in ans:
                ans.append(nums[i])
                nums[count] = ans[count]
                count += 1
        return len(ans)
           

版本2

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        n = len(nums)
        if n == 0:
            return 0
        count = 0
        for i in range(n):
            if nums[i] != nums[count]:
                count += 1
                nums[count] = nums[i]
        return count+1