天天看點

【python-leetcode442-循環排序】數組中重複的資料

問題描述:

給定一個整數數組 a,其中1 ≤ a[i] ≤ n (n為數組長度), 其中有些元素出現兩次而其他元素出現一次。

找到所有出現兩次的元素。

你可以不用到任何額外空間并在O(n)時間複雜度内解決這個問題嗎?

示例:

輸入:

[4,3,2,7,8,2,3,1]

輸出:

[2,3]

class Solution:
    def findDuplicates(self, nums: List[int]) -> List[int]:
        l,r=0,len(nums)-1
        while l<=r:
            if nums[l] != nums[nums[l]-1]:
                nums[nums[l]-1],nums[l]=nums[l],nums[nums[l]-1]
            else:
                l+=1
        res=[]
        for i in range(len(nums)):
            if i+1 != nums[i]:
                res.append(nums[i])
        return res