leetcode原题链接:缺失的第一个正数
上一篇:HOT16-除自身以外数组的乘积
下一篇:HOT18-矩阵置零
题目描述
给你一个未排序的整数数组 nums
,请你找出其中没有出现的最小的正整数。请你实现时间复杂度为 O(n)
并且只使用常数级别额外空间的解决方案。
示例 1:
输入:nums = [1,2,0] 输出:3
示例 2:
输入:nums = [3,4,-1,1] 输出:2
示例 3:
输入:nums = [7,8,9,11,12] 输出:1
解题方法:假设数组的大小为n,则缺失的第一个正数必然为[1, n+1]之间的一个正整数,结果等于n+1只有一种情况,就是数组中的数字是[1, 2, ..., n],其他情况,第一个正整数必然小于或等于n。我们首先对于数组中[1, n]以外的所有小于等于0的元素,我们统一将其大小改写成n + 1(这里只要将其大小改写成大于n的任何一个正整数即可),将不符合解中元素排除后,再针对符合要求的[1, n]中的数字k,我们将数组中第k个位置的值改写成-|nums[k]|,这里负号主要用于标记该数字所在的位置(即k)已经被访问过(即在数组中),为啥改成成-|nums[k]|,用-1标记可以吗?当然不可以,因为原来第k个位置的值也代表一个数字,还需要被访问,当实际访问到第k个元素的时候,-|nums[k]| ==> nums[k]只需要再加个绝对值即可,即nums[k] = | -nums[k] |,所以改成-|nums[k]|的好处是实际访问第k个元素的时候,可以还原成原来加标记前的值,当时改成-1会丢失对应的第k个元素的字面值。
C++代码
#include <iostream>
#include <vector>
#include <cmath> //std::abs
/*解必然是[1, n + 1]之间的数字*/
class Solution {
public:
int firstMissingPositive(std::vector<int>& nums) {
int n = nums.size();
for (int i = 0; i < n; i++) { //先预处理下数组中所有的负数,有负数必然会导致解在[1, n]之间,负数本身不在解的范围之内
if (nums[i] <= 0) {
nums[i] = n + 1;
}
}
// 扫描数组中[1,n]的数字k,给对应num[k-1]的数字前面加个负号标记(为了方便提取数字本身的值)
for (int i = 0; i < n; i++) {
int pos = std::abs(nums[i]);//如果nums[i]之前被打标过,则用|nums[i]|可以提取打标前的数字
if (pos >= 1 && pos <= n ) { //k属于[1,n],给第pos个元素打个标记(加个负号标记)
nums[pos - 1] = -std::abs(nums[pos - 1]);//给第pos个位置数字的值打标
}
}
for (int i = 0; i < n; i++) {
if (nums[i] > 0) {
return i + 1;
}
}
return n + 1;
}
};