leetcode No41. First Missing Positive

Question:

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.

求第一个缺失的正数

Algorithm:

传统的做法,排序然后遍历找到nums[i]!=i+1的元素,复杂度O(nlogn)
可以用bucket sort思想,再开辟一个数组B,把元素作为B的下标存放在数组B中,然后遍历找到nums[i]!=i+1,复杂度为O(n),但是空间复杂度需要O(n)。
我们可以在nums中做类似桶排序的工作,如果nums[i]!=i+1,则交换nums[i]和下标为nums[i]-1的元素,如果nums[i]不在1~n中则不动。

Accepted Code:

class Solution {
public:
    void swap(int &x,int &y)
    {
        int tmp=x;
        x=y;
        y=tmp;
    }
    int firstMissingPositive(vector<int>& nums) {
        int i=0;
        while(i<nums.size())
        {    //nums[i]!=nums[nums[i]-1]是避免元素相等的时候出现死循环
            if(nums[i]!=i+1 && nums[i]>=1 && nums[i]<=nums.size() && nums[i]!=nums[nums[i]-1])  
                swap(nums[i],nums[nums[i]-1]);
            else i++;
        }
        for(int i=0;i<nums.size();i++)
            if(nums[i]!=i+1)
                return i+1;
        return nums.size()+1;
    }
};

20170506更新
滴滴二面考了这题,可惜我忘记了,用hash表做的,空间复杂度O(n),面试官叫我优化,我没优化出来,。。。。就这样挂了。。。。温故而知新。。。惨痛的教训

20170510更新
看了牛客上左程云老师的直播课,又看到这题,他的解法比上面的要更优一些,拿出来分享
设置两个变量,在上面的解法中,如果出现下述情况,说明nums[i]这个数可以扔掉,它要么不在1~n的范围内,要么重复出现,那么我们可以设置r变量,初始为最右及n-1。
如果有上述情况,则nums[l]=nums[r],这样因为nums[r]肯定不是答案,所以r--,待处理的数组范围缩小。
if(nums[i]!=i+1 && nums[i]>=1 && nums[i]<=nums.size() && nums[i]!=nums[nums[i]-1])  

优化后代码如下:
class Solution {
public:
    int firstMissingPositive(vector<int>& nums) {
        int l=0,r=nums.size();
        while(l<r){
            if(nums[l]==l+1)
                l++;
            else if(nums[l]<=l || nums[l]>r || nums[l]==nums[nums[l]-1])
                nums[l]=nums[--r];
            else
                swap(nums[l],nums[nums[l]-1]);
        }
        
        return l+1;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值