1. 程式人生 > 其它 >LeetCode 448——找到所有陣列中消失的數字

LeetCode 448——找到所有陣列中消失的數字

技術標籤: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;
    }
};

四、解題結果