【原地算法】41. 缺失的第一个正数
程序员文章站
2022-04-05 15:41:42
...
问题:
给你一个未排序的整数数组,请你找出其中没有出现的最小的正整数。
例子:
思路
1. 我们将数组中所有小于等于 00 的数修改为 N+1N+1;
2.我们遍历数组中的每一个数 xx,它可能已经被打了标记,
因此原本对应的数为 |x|,其中 ∣∣ 为绝对值符号。如果 ∣x∣∈[1,N],
那么我们给数组中的第 |x| - 1 个位置的数添加一个负号。
注意如果它已经有负号,不需要重复添加;
3.在遍历完成之后,如果数组中的每一个数都是负数,那么答案是 N+1,
否则答案是第一个正数的位置加 11。
代码:
#include <vector>
#include <algorithm>
using namespace std;
class Solution {
private:
int N;
public:
int firstMissingPositive(vector<int>& nums) {
this->N = nums.size();
for (int i = 0; i < N; ++i) {
if(nums[i] <= 0)
nums[i] = N+1;
}
for (int i = 0; i < N; ++i) {
int num = abs(nums[i]);
if (num <= N) {
nums[num - 1] = -abs(nums[num - 1]);
}
}
for (int i = 0; i < N; ++i) {
if(nums[i] > 0)
return i+1; // 返回的是下标+1
}
return N+1;
}
};
上一篇: 你特么不想娶,就明说