Lintcode189/LeetCode41 : First Missing Positive (Bucket Sort经典题)

  1. First Missing Positive
    中文English
    Given an unsorted integer array, find the first missing positive integer.

Example
Example 1:

Input:[1,2,0]
Output:3
Example 2:

Input:[3,4,-1,1]
Output:2
Challenge
Your algorithm should run in O(n) time and uses constant space.

解法1:类似bucket sorting。
这题的意思就是给一个unsorted array,长度N。问从1…N里面那个数字最先missing。要求时间复杂度O(n),空间复杂度O(1)。所以不能用sort, hashmap之类的算法。
基本思想类似count sort,即把每个元素k尽量排到A[k-1]的位置,所以这里必须有一个内部的while loop。
关于这个时间复杂度,为什么是O(n)呢?看了网上很多类似的答案,好像都讲的不是太清楚。我感觉是每个元素都最多只有一次这样的操作(把自己放在该放的那个位置)。有知道的网友麻烦回复一下。不胜感激。

代码如下:

class Solution {
public:
    /**
     * @param A: An array of integers
     * @return: An integer
     */
    int firstMissingPositive(vector<int> &A) {
        int N = A.size();
        if (N == 0) return 1;

        for (int i = 0; i < N; ++i) {
            while(A[i] > 0 && A[i] <= N && A[i] != i + 1) {
                if (A[A[i] - 1] == A[i]) break;  //important!
                swap(A[i], A[A[i] - 1]);
            }
        }
        
        for (int i = 0; i < N; ++i) {
            if (A[i] != i + 1) return i + 1;
        }
        
        return N + 1;
    }
};

解法2:负数标记法。
原理就是把负数和0都标记为n+1,这样就不用考虑它们了。
然后扫描数组,如果遇到绝对值在1…n之内的数,比如说k,则将对应的nums[k-1]变成负数。这样的好处是没有冲掉num[k-1]的值,而且又可以利用其正负值来表示对应位是不是已经有了:负数表示没有,正数表示有。
举例: nums = [2,3,5,-1], n=4
一开始会把-1变成5,也就是nums=[2,3,5,5]。
i=0, 遇到2,其绝对值在1…4内,将nums[2-1]=3变成-3。
i=1, 遇到-3,其绝对值仍然在1…4内,将nums[3-1]=5变成-5。
i=2, 遇到-5, 其绝对值不在1…内,不动
i=3, 遇到5,其绝对值不在1…内,不动
处理完后nums = [2, -3, -5, 5]
最后扫描nums,碰到的第一个>0的元素就是nums[0]=2,所以返回0+1=1。
实际上我们扫描整个数组,还可以发现nums[3]=5>0,所以下一个missing的正整数就是3+1=4。

class Solution {
public:
    int firstMissingPositive(vector<int>& nums) {
        int n = nums.size();
        if (n == 0) return 1;
        for (auto & it : nums) {
            if (it <= 0) it = n + 1;
        }
        for (auto & it : nums) {
            int abs_it = abs(it);
            if (abs_it <= n) nums[abs_it - 1] = -abs(nums[abs_it - 1]);
        }
        for (int i = 0; i < n; ++i) {
            if (nums[i] > 0) return i + 1;
        }
        return n + 1;
    }
};

三刷。用到了额外空间,不算正解。

class Solution {
public:
    int firstMissingPositive(vector<int>& nums) {
        int len = nums.size();
        int minNum = nums[0], maxNum = nums[0];
        int positiveCount = 0;
        for (int i = 0; i < len; i++) {
            if (nums[i] < minNum) minNum = nums[i];
            if (nums[i] > maxNum) maxNum = nums[i];
            if (nums[i] > 0) positiveCount++;
        }
        if (minNum > 1 || maxNum < 0) return 1;
        vector<int> buckets(positiveCount, 0);
        for (int i = 0; i < len; i++) {
            if (nums[i] <= 0) continue;
            if (nums[i] > positiveCount) continue;
            buckets[nums[i] - 1] = 1;
        }
        for (int i = 0; i < positiveCount; i++) {
            if (buckets[i] == 0) return i + 1;
        }
        return positiveCount + 1;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值