Find All Numbers Disappeared in an Array

題目來源
這道題應該算是做過很多次了吧,基本上每次面試都得問一下。
然后剛剛寫代碼還是寫了一會才寫出來。
主要思想就是把相應的位置放上相應的數值。其實相當于利用原有空間來做哈希。

class Solution {
public:
    vector<int> findDisappearedNumbers(vector<int>& nums) {
        auto n = nums.size();
        vector<int> res;
        for (auto i=0; i<n; i++) {
            if (nums[i] != i+1 && nums[nums[i]-1] != nums[i]) {
                swap(nums[i], nums[nums[i]-1]);
                i--;
            }
        }
        for (auto i=0; i<n; i++)
            if (nums[i] != i+1)
                res.push_back(i+1);
        return res;
    }
};

我以為這樣已經是最好的方法了,沒想到還能稍微優化一下。
完全只要遍歷兩遍,一點都不多。
就是第一次遍歷的時候把每個位置里對應數值的位置改為負數。
然后第二遍判斷哪些是正數就是消失的數。
代碼如下:

class Solution {
public:
    vector<int> findDisappearedNumbers(vector<int>& nums) {
        auto n = nums.size();
        vector<int> res;
        for (auto i=0; i<n; i++) {
            int val = abs(nums[i]) - 1;
            if (nums[val] > 0)
                nums[val] = -nums[val];
        }
        for (auto i=0; i<n; i++)
            if (nums[i] > 0)
                res.push_back(i+1);
        return res;
    }
};
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容