Question:
Given an unsorted integer array, find the first missing positive integer.
For example,
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.
求第一个缺失的正数
Algorithm:
传统的做法,排序然后遍历找到nums[i]!=i+1的元素,复杂度O(nlogn)
可以用bucket sort思想,再开辟一个数组B,把元素作为B的下标存放在数组B中,然后遍历找到nums[i]!=i+1,复杂度为O(n),但是空间复杂度需要O(n)。
我们可以在nums中做类似桶排序的工作,如果nums[i]!=i+1,则交换nums[i]和下标为nums[i]-1的元素,如果nums[i]不在1~n中则不动。
Accepted Code:
class Solution {
public:
void swap(int &x,int &y)
{
int tmp=x;
x=y;
y=tmp;
}
int firstMissingPositive(vector<int>& nums) {
int i=0;
while(i<nums.size())
{ //nums[i]!=nums[nums[i]-1]是避免元素相等的时候出现死循环
if(nums[i]!=i+1 && nums[i]>=1 && nums[i]<=nums.size() && nums[i]!=nums[nums[i]-1])
swap(nums[i],nums[nums[i]-1]);
else i++;
}
for(int i=0;i<nums.size();i++)
if(nums[i]!=i+1)
return i+1;
return nums.size()+1;
}
};
20170506更新
滴滴二面考了这题,可惜我忘记了,用hash表做的,空间复杂度O(n),面试官叫我优化,我没优化出来,。。。。就这样挂了。。。。温故而知新。。。惨痛的教训
20170510更新
看了牛客上左程云老师的直播课,又看到这题,他的解法比上面的要更优一些,拿出来分享
设置两个变量,在上面的解法中,如果出现下述情况,说明nums[i]这个数可以扔掉,它要么不在1~n的范围内,要么重复出现,那么我们可以设置r变量,初始为最右及n-1。
如果有上述情况,则nums[l]=nums[r],这样因为nums[r]肯定不是答案,所以r--,待处理的数组范围缩小。
if(nums[i]!=i+1 && nums[i]>=1 && nums[i]<=nums.size() && nums[i]!=nums[nums[i]-1])
优化后代码如下:
class Solution {
public:
int firstMissingPositive(vector<int>& nums) {
int l=0,r=nums.size();
while(l<r){
if(nums[l]==l+1)
l++;
else if(nums[l]<=l || nums[l]>r || nums[l]==nums[nums[l]-1])
nums[l]=nums[--r];
else
swap(nums[l],nums[nums[l]-1]);
}
return l+1;
}
};