- 缺失的第一个正数
给你一个未排序的整数数组 nums ,请你找出其中没有出现的最小的正整数。
请你实现时间复杂度为 O(n) 并且只使用常数级别额外空间的解决方案。
示例 1:
输入:nums = [1,2,0]
输出:3
示例 2:
输入:nums = [3,4,-1,1]
输出:2
示例 3:
输入:nums = [7,8,9,11,12]
输出:1
提示:
1 <= nums.length <= 5 * 105
-231 <= nums[i] <= 231 - 1
题解
看似越简单的题目,操作越复杂,如果不要求时间和空间复杂度,很容易,因为要求的时间复杂度为O(n),所以可以通过哈希的方式,题目要求找到第一个缺失的正整数,那么答案肯定在[1,num.size()+1]之间,于是我们进行哈希,如果某个数字不在这个区间,直接无视,如果在的话,就标记。
时间复杂度为O(n),空间复杂度为O(n)的代码如下:
class Solution {
public:
bool pos[500005];
int firstMissingPositive(vector<int>& nums)
{
memset(pos,0,sizeof(pos));
for(int i=0;i<nums.size();i++)
{
if(nums[i]>=1&&nums[i]<=nums.size())
pos[nums[i]]=true;
}
for(int i=1;i<=nums.size();i++)
if(pos[i]==false)return i;
return nums.size()+1;
}
};
这个很好理解,空间换时间,但是这样也不太好,题目严格要求的是额外的空间复杂度为常数,也就是O(1)这样的。
于是我们想到,直接原地哈希,就是利用原来的nums数组进行上述的标记。
代码如下
class Solution {
public:
int firstMissingPositive(vector<int>& nums)
{
for(int i=0;i<nums.size();i++)
{
while(nums[i]>=1&&nums[i]<=nums.size()&&nums[i]!=nums[nums[i]-1])
{
swap(nums[i],nums[nums[i]-1]);
}
}
for(int i=0;i<nums.size();i++)
if(nums[i]!=i+1)return i+1;
return nums.size()+1;
}
};
这样就没有额外的空间消耗了,符合题目要求。