@author stormma
@date 2017/11/30
生命不息,奋斗不止!
题目
Given an unsorted integer array, find the first missing positive integer.
For example,
Given [1,2,0] return 3,
and [3,4,-1,1] return 2.
Your algorithm should run in O(n) time and uses constant space.
题目分析
题目意思很好理解,找到第一个缺失的整数,对于样例1来说 [1, 2, 0]缺3,样例2 => [1, -1, 3, 4]缺失的是2。
我们通过重新hash元素应在的位置来找出缺失的整数。怎么重新hash元素的位置呢?我们这样定义:
nums[nums[i] - 1] = i
这个是什么意思呢?
很显然,我们用index = 0
来存储整数1,index = 1
来存储整数2,这样答案就很简单了,如果我们没有找到nums[i] != i + 1
的i,那么就直接返回len + 1即可。为什么?比如[1, 2, 3, 4]我们应该返回5。
代码实现
class Solution {
public int firstMissingPositive(int[] nums) {
for (int i = 0; i < nums.length; i++) {
while (nums[i] > 0 && nums[i] <= nums.length && nums[nums[i] - 1] != nums[i]) {
swap(nums, nums[i] - 1, i);
}
}
// find ans
for (int i = 0; i < nums.length; i++) {
if (nums[i] != i + 1) {
return i +1;
}
}
return nums.length + 1;
}
// [3, 4, -1, 1]
// [-1, 4, 3, 1]
// [-1, 1, 3, 4]
// [1, -1, 3, 4]
private void swap(int[] nums, int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
}