题目链接:http://www.lintcode.com/en/problem/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
.
Your algorithm should run in O(n) time and uses constant space.
这个问题用一般的方式解非常容易,无非就是先排序,如果找到一个正整数它与它的下一个正整数不连续,就返回缺了的那个正整数即可
但本题有一个challenge,要求时间复杂度为O(n)且不增加额外的内存,如果没有第二个要求的话,可以再设置一个数组,用以记录那些数字已经出现过,但即使这样都需要遍历这个数组3次,第一次找到数组中的最大值,以确定记录用数组的长度,第二次使用记录数组记录数字的出现情况,第三次找到最小的没有出现过的数字。时间复杂度是满足了,但是增加了额外的内存,而且方法也不够好。
更好的算法当然有,其实上面那种方法已经有更优化的算法的雏形了,说白了就是一个萝卜一个坑,只要把一个数放在它应该出现的地方就可以了。我们将输入的数组的第i个单元的值规定为i+1,只要有i+1这个数,就一定要将这个数放在第i个单元里,除非第i个单元中已经有一个i+1存在。将所有可以放在这个数组规定的某个位置的所有数安放好后,再次遍历整个数组,如果哪个位置放的数并不是规定的值的时候,即表示那个数不存在,返回这个数就好。如果遍历完整个数组都没有发现丢失的数,那么返回这个数组的下一个正整数,如[1,2,0]返回的就是3。
看到这个代码的时候我表示那个while循环太精髓
public int firstMissingPositive(int[] A) {
// write your code here
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;
}
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;
}