LeetCode 448——找到所有陣列中消失的數字
阿新 • • 發佈:2021-01-26
技術標籤:LeetCode刷題
一、題目介紹
給定一個範圍在1 ≤ a[i] ≤ n (n = 陣列大小 ) 的 整型陣列,陣列中的元素一些出現了兩次,另一些只出現一次。
找到所有在 [1, n] 範圍之間沒有出現在陣列中的數字。
您能在不使用額外空間且時間複雜度為O(n)的情況下完成這個任務嗎? 你可以假定返回的陣列不算在額外空間內。
示例:
輸入:
[4,3,2,7,8,2,3,1]
輸出:
[5,6]
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/find-all-numbers-disappeared-in-an-array
二、解題思路
按照題目要求,滿足時間複雜度為O(n),空間複雜度為O(1)的方法如下,原陣列替換法:
(1)將陣列元素對應索引位置的元素值加n。
(2)遍歷修改元素值後的陣列,當某個索引位置i的元素值小於等於n時,則i+1即為小時的數字。
因為陣列中所有元素所對應索引位置的值均增加了n,所以再次遍歷陣列時,如果某個位置的值小於等於n,則證明該索引位置對應數字未出現過,詳見程式碼。
三、解題程式碼
class Solution { public: vector<int> findDisappearedNumbers(vector<int>& nums) { vector<int> vec; int n = nums.size(); if(n == 0) return vec; for(int i = 0; i < n; ++i) { int index = (nums[i] - 1)%n; nums[index] += n ; } for(int i = 0; i < n; ++i) { if(nums[i] <= n) vec.push_back(i+1); } return vec; } };