[LeetCode] First Missing Positive

Given an unsorted integer array, find the first missing positive integer.

For example,
Given [1,2,0] return 3,
and [3,4,-1,1] return 2.

Your algorithm should run in O(n) time and uses constant space.

參考這里的解答：http://dl.dropbox.com/u/19732851/LeetCode/FirstMissingPositive.html

主要的思想就是把對應的數放到對應的索引上，例如1放到A[1]上，這樣只需要O(n)的遍歷就能完成，然后用一個O(n)的遍歷找第一個沒有放到索引上的數返回。

最后就是可能A[0] == n，這時就要有個特殊情況的處理。

class Solution {
public:
    int firstMissingPositive(int A[], int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        int i = 0;
        while(i < n)
        {
            if (A[i] == i)
                i++;
            else
            {
                if (0 <= A[i] && A[i] < n && A[A[i]] != A[i])
                {
                    int t = A[i];
                    A[i] = A[A[i]];
                    A[t] = t;
                    continue;
                }
                else
                    i++;
            }
        }
        
        for(int i = 1; i < n; i++)
            if (A[i] != i)
                return i;
        
        return A[0] == n ? n + 1 : n;
    }
};