LeetCode 448. 找到全部數組中消失的數字

給定一個範圍在  1 ≤ a[i] ≤ n ( n = 數組大小 ) 的 整型數組，數組中的元素一些出現了兩次，另外一些只出現一次。

找到全部在 [1, n] 範圍之間沒有出如今數組中的數字。

您能在不使用額外空間且時間複雜度爲O(n)的狀況下完成這個任務嗎? 你能夠假定返回的數組不算在額外空間內。

示例:

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

輸出:
[5,6]

 

class Solution {
public:
    int abs(int x) {
        return x > 0 ? x : -x;
    }

    vector<int> findDisappearedNumbers(vector<int>& nums) {
        if (nums.empty())
            return {};
        
        int idx;
        for (int i = 0; i < nums.size(); ++i)
        {
            idx = abs(nums[i]) - 1;
            if (nums[idx] > 0)
                nums[idx] = -nums[idx];
        }
        vector<int> res;
        for (int i = 0; i < nums.size(); ++i)
        {
            if (nums[i] > 0)
                res.push_back(i + 1);
        }
        return res;
    }
};