1. 程式人生 > >Leetcode篇:刪除重復元素

Leetcode篇:刪除重復元素

moved 函數 remove type software 給定 swe 排序 條件


@author: ZZQ
@software: PyCharm
@file: removeDuplicates.py
@time: 2018/9/23 13:51
要求:
給定一個排序數組,你需要在原地刪除重復出現的元素,使得每個元素只出現一次,返回移除後數組的新長度。
不要使用額外的數組空間,你必須在原地修改輸入數組並在使用 O(1) 額外空間的條件下完成。
e.g.:
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。
class Solution():
    def __init__(self):
        pass

    def removeDuplicates(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if not nums:
            return 0
        index = 0
        current_value = nums[0]
        new_len = 1
        for i in range(1, len(nums)):
            if nums[i] > current_value:
                index += 1
                nums[index] = nums[i]
                new_len += 1
            current_value = nums[i]
        return new_len


if __name__ == "__main__":
    answer = Solution()
    nums = []
    print(answer.removeDuplicates(nums=nums))
    print nums

Leetcode篇:刪除重復元素