[LeetCode]缺失的第一个正数
给你一个未排序的整数数组 nums ,请你找出其中没有出现的最小的正整数。
进阶:你可以实现时间复杂度为 O(n) 并且只使用常数级别额外空间的解决方案吗?
示例 1:
输入:nums = [1,2,0] 输出:3 示例 2:
输入:nums = [3,4,-1,1] 输出:2 示例 3:
输入:nums = [7,8,9,11,12] 输出:1
提示:
0 <= nums.length <= 300
-231 <= nums[i] <= 231 - 1
这道题刚开始思考的时候,我打算先将数组中的数进行排序,然后再遍历就可以找到缺失的第一个正数,但是我们所知道的排序算法的时间复杂度最小都是O(nlog2n),明显不符合题中要求的时间复杂度O(n)。
我又想到用哈希表来处理,我们可以将数组所有的数放入哈希表,随后从 1 开始依次枚举正整数,并判断其是否在哈希表中。这样做如果数组的长度为 N,时间复杂就为 O(N),空间复杂度为 O(N),那么空间复杂度又不符合题意中的常数级。
哦豁,这可怎么办,查看了各位大佬的题解,发现可以这样来解决这道题。
实际上,对于一个长度为 N 的数组,其中没有出现的最小正整数只能在 [1, N+1]中。
这是因为如果 [1, N]都出现了,那么答案是 N+1,否则答案是 [1, N]中没有出现的最小正整数。
这样一来,我们将所有在 [1, N] 范围内的数放入哈希表,也可以得到最终的答案。而给定的数组恰好长度为 N,这让我们有了一种将数组设计成哈希表的思路:
我们对数组进行遍历,对于遍历到的数 x,如果它在 [1, N] 的范围内,那么就将数组中的第 x-1个位置(注意:数组下标从 0 开始)打上「标记」。在遍历结束之后,如果所有的位置都被打上了标记,那么答案是 N+1,否则答案是最小的没有打上标记的位置加 1。
那么如何设计这个「标记」呢?由于数组中的数没有任何限制,因此这并不是一件容易的事情。但我们可以继续利用上面的提到的性质:由于我们只在意 [1, N]中的数,因此我们可以先对数组进行遍历,把不在 [1, N] 范围内的数修改成任意一个大于 N 的数(例如 N+1)。这样一来,数组中的所有数就都是正数了,因此我们就可以将「标记」表示为「负号」。算法的流程如下:
我们将数组中所有小于等于 0 的数修改为 N+1;
我们遍历数组中的每一个数 x,它可能已经被打了标记,因此原本对应的数为 |x|,其中 ∣∣ 为绝对值符号。如果∣x∣∈[1,N],那么我们给数组中的第 |x| - 1个位置的数添加一个负号。注意如果它已经有负号,不需要重复添加;
在遍历完成之后,如果数组中的每一个数都是负数,那么答案是 N+1,否则答案是第一个正数的位置加 1。
以上时官方给出的题解思路,举个例子来说明一下加深理解。
nums = [3,4,-1,1]
- 那么我们对这个数组首先进行预处理,将数组中0,负数全部设置为n+1;处理后 数组为: nums = [3,4,5,1]
- 然后我们开始遍历进行标记
- 遍历第一个数为3,我们把下标为2(3-1=2,数组下标从0开始)的数字5标记为-5
- 遍历第二个数4,我们把下标为3(4-1=3,数组下标从0开始)的数字1标记为-1
- 遍历第三个数为-5,这个数字已经被打了标记,原本对应的数为|-5|=5(加绝对值),我们发现5不在范围 [1, N] 范围内所以不进行标记处理。
- 遍历第四个数,我们把下标为0(1-1=0,数组下标从0开始)的数字3标记为-3
- 所以现在标记处理后的数组为:nums = [-3,4,-5,-1]
- 我们遍历发现有一个正数,那么答案就是第一个正数的数组下标位置+1,即1+1=2,所以数组nums = [3,4,-1,1] 缺失的第一个正数是2。
然后我们就可以开始编写代码了(Java代码)
class Solution {
public int firstMissingPositive(int[] nums) {
int n = nums.length;
for (int i = 0; i < n; ++i) {
if (nums[i] <= 0) {
nums[i] = n + 1;
}
}
for (int i = 0; i < n; ++i) {
int num = Math.abs(nums[i]);
if (num <= n) {
nums[num - 1] = -Math.abs(nums[num - 1]);
}
}
for (int i = 0; i < n; ++i) {
if (nums[i] > 0) {
return i + 1;
}
}
return n + 1;
}
}
执行通过。