题意:
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)时间,O(1)空间。
那么要怎么做呢,
假设数组的第i项对应值 i + 1,即 nums[0] 应对应 1.
刚开始我想到:可以遍历一遍,如果发现当前数应该放在数组的其他位置,就交换当前位置 i 和对应位置 nums[i] 的两个数。这样做完之后,再遍历一遍,如果当前位置和当前数不对应,就说明缺少这个数。
这种做法有个问题,比如序列:3 4 -1 1
首先变为-1 4 3 1,然后变为-1 1 3 4
这样的做法导致后面的较小数无法放到对应位置。
然后我想到应该用while循环做交换,一直交换到所有有对应位置的数都能到达对应位置,这样就成功解决了这道题。
class Solution {
public:
int firstMissingPositive(vector<int>& nums) {
if(nums.empty()) return 1;
for(int i = 0; i < nums.size(); ++i){
if(nums[i] - 1 != i){
while(nums[i] >= 1 && nums[i] <= nums.size() && nums[nums[i] - 1] != nums[i]){
int tmp = nums[nums[i] - 1];
nums[nums[i] - 1] = nums[i];
nums[i] = tmp;
}
}
}
for(int i = 0; i < nums.size(); ++i)
if(nums[i] - 1 != i)
return i + 1;
return nums.size() + 1;
}
};