[LeetCode] 找到第一个不存在的正整数 First Missing Positive

给定一个未排序的整数数组，找到最小的且不存在于该数组的正整数。

例如，[-1, 2, 4, 3]中，1是最小的且不存在于该数组的正整数。

思路：

对于数组中的每一个元素arr[i]，如果 0 < arr[i] <= n 且 arr[i] != i+1，那么我们就要把 arr[i] 移动到 i+1 的位置上去，其中，n是数组中元素的个数。

最后，对数组 arr 重新扫描一边，如果 arr[i] != i+1, i+1 就是所求的正整数。

下面的代码有bug：当输入时[1,1]的时候，第9行的while-loop会陷入无限循环。

#include <iostream>
using namespace std;

int solve(int arr[], int n)
{
        for(int i=0; i<n; i++)
                if(arr[i]==i+1) continue;
                else if(arr[i]>0 && arr[i]<=n)
                        while(arr[i]!=i+1)
                        {
                                swap(arr[i], arr[arr[i]-1]);
                                if(arr[i]<=0 || arr[i]>n) break;
                        }
      
        for(int i=0; i<n; i++)
                if(arr[i]!=i+1)
                        return i+1;
}       
        
int main(int argc, char** argv)
{           
        int arr[] = {-1,4,3,1}; 
        int missing = solve(arr, sizeof(arr)/sizeof(int));
        cout<<missing<<endl;
}   

下面是正确无误的代码。代码的第12行A[i] == A[A[i] - 1]对于[1,1]这种有重复元素的数组很重要。：

int firstMissingPositive(int A[], int n) {
    bucket_sort(A, n);
    for (int i = 0; i < n; ++i)
        if (A[i] != (i + 1))
    return i + 1;
    return n + 1;
}

void bucket_sort(int A[], int n) {
    for (int i = 0; i < n; i++) {
        while (A[i] != i + 1) {
            if (A[i] <= 0 || A[i] > n || A[i] == A[A[i] - 1])
                break;
            swap(A[i], A[A[i] - 1]);
        }
    }
}

相关问题：Find duplicates in an array