问题描述
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.
解题思路
问题思路比较简单,即找出缺失的正数、但是问题要求只能用常数额外空间,即不能再另外开辟非常数级的额外空间,所以我们可以在输入数组上就地进行swap操作。我们可以这样思考:交换数组元素,使得数组中第i位存放数值(i+1)。最后遍历数组,寻找第一个不符合此要求的元素,返回其下标。这样只要两次遍历数组,时间复杂度为O(n)。
代码展示
#include <iostream>
#include <stdlib.h>
#include <vector>
#include <string>
using namespace std;
class Solution {
public:
int firstMissingPositive(vector<int>& nums){
int length = nums.size();
int i = 0;
while(i < length){ //将正整数 a放到第a-1的位置上
if(nums[i] != (i+1)&& nums[i] >= 1&&nums[i] <= length && nums[nums[i]-1] != nums[i])
swap(nums[i], nums[nums[i]-1]);
else
i++;
}
for (i = 0; i < length; ++i) //与位置不对应的即为缺失的正数
if (nums[i] != (i+1))
return i+1;
return length+1;
}
};
int main(){
int n;
vector<int> nums;
cout<<"请输入向量长度:";
cin>>n;
int num[n];
for(int i = 0;i<n;i++){
cin>>num[i];
nums.push_back(num[i]);
}
Solution solution;
int result=solution.firstMissingPositive(nums);
cout<<result<<endl;
}
运行结果展示