给定一个未排序的整数数组,找到最小的且不存在于该数组的正整数。
例如,[-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]);
}
}
}