[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;
    }
};