【Lintcode】189. First Missing Positive

题目地址：

https://www.lintcode.com/problem/first-missing-positive/description

给定一个数组，返回$(0,1,2,3,...)$中第一个未出现在数组中的数。

思路是，通过不停的交换，尽量将值为$i$的数移到下标为$i-1$的位置上去，然后从左到右扫描即可。举例：对于$(3,4,-1,1)$，我们先把$3$交换到数组第$3$个位置上去，得到$(-1,4,3,1)$，这样$3$这个数字就归位了，接下来$-1$没有地方可以去，就直接略过，接下来把$4$交换到数组第$4$个位置上去，得到$(-1,1,3,4)$，$4$已经归位，再把换过来的$1$归位，得到$(1,-1,3,4)$，接下来的扫描都不会产生交换。最后从左到右扫描看哪个数字没出现即可。代码如下：

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]代表的数字“应该”在的位置上的数字不等于A[i]，就将其换过去
            while (1 <= A[i] && A[i] <= A.length && A[A[i] - 1] != A[i]) {
                swap(A, A[i] - 1, i);
            }
        }
    
        for (int i = 0; i < A.length; i++) {
            if (A[i] != i + 1) {
                return i + 1;
            }
        }
        // 如果上面的for循环里没有return，说明每个数字都归位了，
        // 那第一个missing的正数就是A.length + 1
        return A.length + 1;
    }
    
    private void swap(int[] A, int i, int j) {
        int tmp = A[i];
        A[i] = A[j];
        A[j] = tmp;
    }
}

时间复杂度$O(n)$，空间$O(1)$。

算法正确性证明：
设数组长度为$l$，那么在循环到某个位置时，做的swap的次数一定是有限的，且不会超过$l$，这是因为每swap一次，就会有一个数字归位，归位后那个数字就不会再被swap到了，所以最多的情况就是数组每个位置都swap了一遍，也就是次数不超过$l$。所以while循环一定会退出，也就是说第一个for循环一定会在有限时间内结束。接下来第二个for循环遍历到的第一个$A[i]\ne i+1$的数，说明$i+1$在这个数组中不存在，所以返回$i+1$。如果遍历到末尾说明第一个missing正数是$l+1$。

时间复杂度证明：
只需注意到swap最多只会执行不超过$l$次即可。因为每swap一次都会使得一个数字归位，而最多只能有$l$个数字归位。