Leetcode 41. First Missing Positive

来源:互联网 发布:js两个对象数组合并 编辑:程序博客网 时间:2024/06/04 23:20

@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;    }}
原创粉丝点击