lintcode-medium-First Missing Positive

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

 

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

 

思路：

通过在数组进行的交换，把A[i]赋值为i + 1

再遍历数组，找到第一个A[i] 不等于 i + 1的位置。如果没有这样的数，则缺少的正数是A中最后一个数的下一个，即A.length + 1

public class Solution {
    /**    
     * @param A: an array of integers
     * @return: an integer
     */
    public int firstMissingPositive(int[] A) {
        // write your code here    
        if(A == null || A.length <= 0)
            return 1;
        
        for(int i = 0; i < A.length; i++){
            
            while(A[i] > 0 && A[i] <= A.length && A[i] != i + 1){
                int temp = A[A[i] - 1];
                if(temp == A[i]){
                    break;
                }
                else{
                    A[A[i] - 1] = A[i];
                    A[i] = temp;
                }
            }
            
        }
        
        for(int i = 0; i < A.length; i++)
            if(A[i] != i + 1)
                return i + 1;
        
        return A.length + 1;
    }
}