题目如下:
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.
题目大意如下:
给定一个为排序的数组,找出第一个缺失的正整数。
例如给定数组[1, 2, 0] 返回3;
给定数组[3, 4, -1, 1] 返回2;
算法时间复杂度要求为O(n),空间复杂度为O(1).
对于此题很容易想到HashMap的方法,直接用一位数组flags,用数组下标对应正整数。首先遍历一遍数组,用数组flags记录正整数i是否出现过,出现过就置为1,未出现就为0。然后再便利数组flags,查看哪个数为出现,即为第一个缺失的正整数。此算法的时间复杂度为O(n),但空间复杂度为O(n),不符合题意。
此处采用另外一种方法。
我们还是从前往后遍历数组,对于数nums[i], 我们需要讲nums[i]放到正确的位置上,由于正整数从1开始,故对于数nums[i]大于0,则需将nums[i]放到数组nums下标为nums[i]的位置。经过一次遍历数组后,数都大致放到了自己的正确的位置上。然后再遍历一遍数组,找出第一个位置上放的数不正确的位置i,即为第一个缺失的数。如果所有位置i上的数都正确,则返回下一个应该出现的正整数,即nums.size() + 1;
代码如下:
int firstMissingPositive(vector<int> &nums) {
if(nums.size() == 0) return 1;
for(int i = 0; i < nums.size(); ++i) {
while(nums[i] > 0 && nums[i] != i + 1
&& nums[i] < nums.size() &&
nums[nums[i] - 1] != nums[i])
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;
}