1. 程式人生 > >【leetcode】#陣列【Python】80. Remove Duplicates from Sorted Array II 刪除排序陣列中的重複項 II 雙指標

【leetcode】#陣列【Python】80. Remove Duplicates from Sorted Array II 刪除排序陣列中的重複項 II 雙指標

連結:

題目:

給定一個排序陣列,你需要在原地刪除重複出現的元素,使得每個元素最多出現兩次,返回移除後陣列的新長度。

不要使用額外的陣列空間,你必須在原地修改輸入陣列並在使用 O(1) 額外空間的條件下完成。

示例 1:

給定 nums = [1,1,1,2,2,3],

函式應返回新長度 length = 5, 並且原陣列的前五個元素被修改為 1, 1, 2, 2, 3 。

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

我的解法1:滿足條件pop掉

class Solution(object):
    def removeDuplicates(self, nums):
        if
nums == []: return 0 if len(nums) <= 2: return len(nums) i = 2 while i < len(nums): if nums[i] == nums[i-1] == nums[i-2]: nums.pop(i) else: i += 1

我的解法2: 雙指標 nums[j] > nums[i-2]時,賦值

class Solution(object):
    def removeDuplicates(self, nums):
        i,j = 0,0
        while j < len(nums):           
            if i<2 or nums[j]>nums[i-2]:
                nums[i] = nums[j]
                i += 1
                j += 1
            else:
                j+=1
        return
i

別人的解法:相當於雙指標,一個指標一直走,一個指標指向賦值的位置,賦值的情況是nums[j] > nums[i-2],實現的很簡潔

class Solution(object):
    def removeDuplicates(self, nums):
        i = 0
        for n in nums:
            if i < 2 or n > nums[i-2]:
                nums[i] = n
                i += 1
        return i