缺失的正整数,我自己解出来了,和官方题解思路不一样
题目描述:
给你一个未排序的整数数组,请你找出其中没有出现的最小的正整数。
示例 1
输入: [1,2,0]
输出: 3
示例 2:
输入: [3,4,-1,1]
输出: 2
示例 3:
输入: [7,8,9,11,12]
输出: 1
提示:
你的算法的时间复杂度应为O(n),并且只能使用常数级别的额外空间。
我的思路很直接,排序之后,重复的去重,剩下的第一个元素与1比较,第二个元素与2比较,第三个元素与3比较,依次类推,只要出现小于的情况,直接返回结果。
public int firstMissingPositive(int[] nums) {
if(null == nums) return 1;
Arrays.sort(nums); // 排序
int compare = 0;
for(int i = 0; i < nums.length; i++){
if(nums[i] <= 0) continue;
if(i > 0 && nums[i-1] == nums[i]) continue; // 重复的数字直接略过
compare++;
if(compare < nums[i]) return compare;
}
// 数组中出现的正数,排序后恰好是递增的,那就返回compare+1
return compare+1;
}
我的解法,没什么特别之处,时间复杂度是O(n),空间复杂度是常数级别的,符合题目要求。
下面看下官方的题解,思路很巧妙,我压根就想不到。
若数组中有 n 个元素,且从1开始递增的,那答案一定是n+1,否则答案一定是大于0小于n,
第一步,把数组中的元素,小于0的,大于n+1的都替换为n+1;
第二步,对数组下标作文章,遍历数组,比如第一个元素值是5,那就把第5个元素设置为负数
第三步,再次遍历数组,第一个大于0的数,说明没有给它做标记的,即那个数字没出现。
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;
}
官方的题解确实很优雅,我真的想不到。我的解法,也还行,毕竟是符合题目要求的。