缺失的第一个正数
leetcode 41. 缺失的第一个正数
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/first-missing-positive
题目描述
给你一个未排序的整数数组 nums ,请你找出其中没有出现的最小的正整数。
请你实现时间复杂度为 O(n) 并且只使用常数级别额外空间的解决方案。
示例 1:
输入:nums = [1,2,0]
输出:3
示例 2:
输入:nums = [3,4,-1,1]
输出:2
示例 3:
输入:nums = [7,8,9,11,12]
输出:1
提示:
1 <= nums.length <= 5 * 105
-231 <= nums[i] <= 231 - 1
原地哈希
令数组长度为 n,那么答案必然在 [1, n + 1] 范围内。
因此我们可以使用「桶排序」的思路,将每个数放在其应该出现的位置上。
基本思路为:
按照桶排序思路进行预处理:保证 1 出现在 nums[0] 的位置上,2 出现在 nums[1] 的位置上,…,n 出现在 nums[n - 1] 的位置上。不在 [1, n] 范围内的数不用动。
例如样例中 [3,4,-1,1] 将会被预处理成 [1,-1,3,4]。
遍历 nums,找到第一个不在应在位置上的 [1, n] 的数。如果没有找到,说明数据连续,答案为 n + 1
例如样例预处理后的数组 [1,-1,3,4] 中第一个 nums[i] != i + 1 的是数字 2(i = 1)。
代码演示
/**
* 桶排序,将1 - n 的数字排到相应的位置上
* @param nums
* @return
*/
public int firstMissingPositive(int[] nums) {
int n = nums.length;
for (int i = 0;i < nums.length;i++){
//在1 到 n 范围上的数字进行排序
// nums[i] > 0 nums[i] <= n
while (nums[i] > 0
&& nums[i] <= n
&& nums[i] != i + 1
&& nums[i] != nums[nums[i] - 1]){
swap(nums,i,nums[i] - 1);
}
}
for (int i = 0; i < n;i++){
if (nums[i] != i + 1){
return i + 1;
}
}
return n + 1;
}
/**
* 交换
* @param nums
* @param i
* @param j
*/
public void swap(int[]nums,int i,int j){
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}