LeetCode: First Missing Positive

Problem:

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://blog.unieagle.net/?p=594。

1. 第一次遍历，设置所有的非正整数为一个无关紧要的数，如n+2，因为其永远不可能为解。
2. 第二次遍历，用第i位数的正负来表示数i+1是否在数组A[]中。
3. 第三次遍历，如遇到A[i]为整数，则i+1不在数组A[]中，如果1~n都在，则返回n+1。

class Solution {
public:
    int firstMissingPositive(int A[], int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        if (n == 0) return 1;
        for (int i = 0; i < n; ++i)
            if (A[i] <= 0)
                A[i] = n + 2;
        
        for (int i = 0; i < n; ++i)
        {
            int idx = abs(A[i]) - 1;
            if (idx < n && idx >= 0)
            {
                if (A[idx] > 0) A[idx] = ~A[idx] + 1;
            }
        }
        
        for (int i = 0; i < n; ++i)
        {
            if (A[i] >= 0)
                return i + 1;
        }
        return n + 1;
    }
};