如果没有限制,解题很简单,哈希表就可以解决,但那不能达到目的。
同时我们也很容易想到排序,但排序最低是O(nlogn)的时间复杂度,那我们可以另辟蹊径。
考虑到题目条件,可以想到:我们要找的数一定是[1,n+1]这个范围(n为数组长度)
所以只需要让元素回到对应的位置即可
class Solution {
public:
int firstMissingPositive(vector<int>& nums) {
int n=nums.size(),i;
for(i=0;i<n;i++){
if(nums[i]<=n&&nums[i]>=1&&nums[i]!=i+1)
{
swap(nums[i],nums[nums[i]-1]);
}
}
for(i=0;i<n;i++){
if(nums[i]!=i+1)
return i+1;
}
return n+1;
}
};
这样做符合要求,O(n)的时间复杂度和O(1)的空间复杂度
但明显是不对的,考虑这个样例
nums = [0,1,2]
那么我们只能循环直到不符合特殊情况,只需要把if改成while
同时里面增加一个判断避免死循环,如 nums = [1,1]
注意 不能用 nums[i]-1!=i 要用 nums[i] != i+1 不然会超出int的范围
AC 代码:
class Solution {
public:
int firstMissingPositive(vector<int>& nums) {
int n=nums.size(),i;
for(i=0;i<n;i++){
while(nums[i]<=n&&nums[i]>=1&&nums[i]!=i+1)
{
if(nums[nums[i]-1]==nums[i]) break;
swap(nums[i],nums[nums[i]-1]);
}
}
for(i=0;i<n;i++){
if(nums[i]!=i+1)
return i+1;
}
return n+1;
}
};
两种情况是自己调的,但套了两个循环实在不觉得还是O(n)的时间复杂度
看了官方题解发现确实是这样做......
如果有更好的解法望告知 orz