由于只有 n 个数,因此不用担心 负数和大于 n 的数,所以
- 当遇见负数或大于 n 的数时,将其置为 1,此时列表中数均满足 [1, n];
- 对于列表中的数 nums[i],将其作为 index 置 nums[ nums[i] ]为负数,则表示 nums[i] 出现过;
- 这样处理后,重现遍历,则遇见的正数则表示未出现过;
- 注意数组大小为 0 ~ n-1,所以当值为 n 时,对 0 处进行处理;
- 另外注意的是因为第一步会将 负数和大于 n 的数置为 1,这样在第二步时 1 必然会被置为负数,因此我们需要在刚开始排除结果是 1 的情形;
详见代码:
class Solution {
public:
int firstMissingPositive(vector<int>& nums) {
if(nums.size() == 0){
return 1;
}
if(nums.size() == 1){
return (nums[0] == 1 ? 2 : 1);
}
int n = nums.size();
// 排除 1 的情况
int cnt = 0;
for(int i = 0; i < n; i++){
if(nums[i] == 1){
cnt++;
break;
}
}
if(cnt == 0){
return 1;
}
// 所有负数和不在 [1,n] 之间的数置为 1
for(int i = 0; i < n; i++){
if(nums[i] <= 0 || nums[i] > n){
nums[i] = 1;
}
}
// 将所有数的值 nums[i] 对应的下标里的数 nums[nums[i]]置为负数,表示正数 nums[i] 存在
for(int i = 0; i < n; i++){
int a = fabs(nums[i]);
if(a == n){
nums[0] = -fabs(nums[0]);
}
else{
nums[a] = -fabs(nums[a]);
}
}
// 若存在正数 nums[i],则说明正数 i 不存在
for(int i = 1; i < n; i++){
if(nums[i] > 0){
return i;
}
}
return (nums[0] > 0 ? n : n+1);
}
};