LintCode 丟失的第一個正整數

丟失的第一個正整數
給出一個無序的正數數組，找出其中沒有出現的最小正整數。
樣例
如果給出 [1,2,0], return 3 如果給出 [3,4,-1,1], return 2
挑戰
只允許時間複雜度O(n)的算法，並且只能使用常數級別的空間。

Solution:
方法一：一種最直接的方法就是對數組進行排序然後再查找.

class Solution {
public:
    /**    
     * @param A: a vector of integers
     * @return: an integer
     */
    int firstMissingPositive(vector<int> A) {
        // write your code here
        sort(A.begin(), A.end());
        int data = 0;
        for (int i = 0; i < A.size(); ++i) {
            if (A[i]>0 && A[i]-data > 1)
                return data +1;
            else if (A[i] > 0)
                data = A[i];
        }
        return data + 1;
    }
};

方法二：轉自LeetCode discuss
idea：通過交換元素的位置是的在正確的位置存放正確的數字(例如給定一個數字3那麼他應該在第三個位置,下標爲2)，當存在重複的數字時將數字移到數組的後面：
例如：給出[3, 4, -1, 1]

對於給定具有重複數字的也是一樣

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