給你一個含 n 個整數的數組 nums ,其中 nums[i] 在區間 [1, n] 內。請你找出所有在 [1, n] 范圍內但沒有出現在 nums 中的數字,并以數組的形式返回結果。
示例 1:
輸入:nums = [4,3,2,7,8,2,3,1] 輸出:[5,6]
示例 2:
輸入:nums = [1,1] 輸出:[2]
提示:
n == nums.length1 <= n <= 1051 <= nums[i] <= n
進階:你能在不使用額外空間且時間復雜度為 O(n) 的情況下解決這個問題嗎? 你可以假定返回的數組不算在額外空間內。
實現:
可以使用原地修改數組的方法,在遍歷 nums 數組時,將出現的數字作為下標,將對應位置的數字取相反數,即 nums[abs(nums[i])-1] = -abs(nums[abs(nums[i])-1])。
遍歷完成后,再次遍歷數組,將為正數的下標加 1 加入到結果數組中即可。
時間復雜度為 O(n),空間復雜度為 O(1)。
以下是 Python 代碼實現:
class Solution:
def findDisappearedNumbers(self, nums: List[int]) -> List[int]:
res = []
for i in range(len(nums)):
index = abs(nums[i]) - 1
if nums[index] > 0:
nums[index] = -nums[index]
for i in range(len(nums)):
if nums[i] > 0:
res.append(i+1)
return res
浙公網安備 33010602011771號