题目:
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.
题意:
给定一个未排序的数组,找出第一个缺失的正数。
思路:
这道题目要求O(n)时间复杂度,并且要求常量的空间复杂度。我们只要扫描两次数组,第一次扫描的时候把i元素放到下标是i-1的数组中的位置,如果这个数组超过了数组的大小或者小于1,那么不去考虑它的位置。我们只考虑1-n的正数放在对应的0-n-1的下标处。第二次扫描数组的时候找到第一个不等于下标加一的位置,然后返回这个下标加一。当然也有可能数组原来的元素就是1~n,那么扫描到数组末尾时也没找到不满足条件的元素。那么返回n+1。
这里比较麻烦的地方是替换元素。比如扫描到第k个下标,对应的元素是nums[k],如果1=< nums[k] <=size并且nums[k] != k+1,那么需要将nums[k]放到它应该待的位置,它应该待的位置就是下标nums[k] - 1,但是如果nums[k] - 1下标对应的数字已经是正确的数字即nums[k],那么这个时候就不要再去把这个元素放到nums[k] - 1的位置。当把nums[k]放到nums[k]-1的位置上的时候,我们需要将原来nums[k]-1位置上的元素放到k位置上,循环交换,交换的停止条件是:
- nums[k] < 1 || nums[k] > size
- nums[k] = k + 1
以上。
代码如下:
class Solution {
public:
int firstMissingPositive(vector<int>& nums) {
int size = nums.size();
for (int i = 0; i < size; i++) {
int j = i;
while (nums[j] > 0 && nums[j] < size && nums[j] != j + 1 && nums[nums[j] - 1] != nums[j]) {
swap(nums[nums[j] - 1], nums[j]);
}
}
for (int i = 0; i < size; i++)
if (nums[i] != i + 1)return i + 1;
return size + 1;
}
};