1. 程式人生 > 其它 >【陣列】力扣442:陣列中重複的資料

【陣列】力扣442:陣列中重複的資料

給你一個長度為 n 的整數陣列 nums ,其中 nums 的所有整數都在範圍 [1, n] 內,且每個整數出現 一次 或 兩次 。請你找出所有出現 兩次 的整數,並以陣列形式返回。
你必須設計並實現一個時間複雜度為 O(n) 且僅使用常量額外空間的演算法解決此問題。
示例1:

輸入:nums = [4,3,2,7,8,2,3,1]
輸出:[2,3]

示例2:

輸入:nums = [1]
輸出:[]

思想:
由於數字i在1到n之間,可以將nums[i]設定為-nums[i],以此作為已經遍歷過一次的標記。那麼,如果遍歷到某個i發現nums[i]<0,則說明這個值已經出現過一次了,就是結果。

class Solution:
    def findDuplicates(self, nums: List[int]) -> List[int]:
        res = []
        for i in nums:
            # 取絕對值
            i = abs(i)
            # 把相應下標減1的值設為負數
            if nums[i-1] > 0:
                nums[i-1] *= -1
            # 值為負的話,說明該值已經出現過,新增到輸出列表l中
            else:
                res.append(i)
        return res