leetcode problem 41 -- First Missing Positive

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,2,4,5的第一个缺失的正整数是3.要求时间复杂度为o(n) 空间复杂度为o(1)

思路:

  第一种是排序,时间复杂度为O(nlgn) 不行

  第二种是hash查找  对每一个数字都用额外空间找到正确的位置. 然后一次遍历找到第一个缺失的正整数.  时间复杂度为O(n)  但是空间复杂度为O(n) 也不符合题目要求

  第三种直接在输入数组上进行交换操作, 使得每个数字通过交换能在正确的位置上.如3,1,7,2 这组数字第一次交换后得到7,1,3,2   数字3放置在了正确的位置.

    这种时间复杂度为O(n)  而且只需要常数辅助空间.  但我认为这种方法依然不复合题目要求,因为它破坏了原始数组.  它其实也类似于hash, 需要O(n)的空间,只不过这O(n)的空间借用了原始输入数组上的.

  目前我还没找到即不破坏原始数组,空间复杂度又为O(1)的方法

 

代码1(hash法):

  

class Solution {
public:
    int firstMissingPositive(vector<int>& nums) {
        if (nums.size() == 0)
            return 1;
        auto it_max = std::max_element(nums.begin(), nums.end());    
        auto it_min = std::min_element(nums.begin(), nums.end());

        vector<bool> vec(*it_max - *it_min, false);
        for (auto elem : nums) 
            vec[elem - *it_min] = true;
        for (auto it = 1; it <= *it_max; ++it) {
            if (it < *it_min || (it > 0 && !vec[it - *it_min]))
                return it;
        }    
        return *it_max > 0 ? *it_max + 1 : 1;
    }

private:
};

 

 

 

代码2(直接在原数组上进行交换, 空间代价为o(1))

class Solution {
public:
    int firstMissingPositive(vector<int>& nums) {
        if (nums.empty())
            return 1;
        auto begin = getFirstPositivePos(nums);
        if (*begin < 0)
            return 1;
        for (auto it = begin; it != nums.end(); ) {
            auto realPos = begin + *it - 1;
            if (realPos < nums.end() && realPos != it && *realPos != *it)
                iter_swap(it, realPos);
            else
                ++it;
        }
        int index = 1;
        for (auto it = begin; it != nums.end(); ++it, index++) {
            if (index != *it)
                return index;
        }
        return index;
    }

private:
    vector<int>::iterator getFirstPositivePos(vector<int>& nums) {
        auto first = nums.begin();
        auto last = nums.end()-1;
        while (true) {
            while (first < last && *first <= 0 ) 
                ++first;
            while (first < last && *last > 0)
                --last;
            if (first < last)
                iter_swap(first, last);    
            else
                break;
        }
        return first;
    }
};

 

posted on 2015-05-05 15:12  lysuns  阅读(196)  评论(0编辑  收藏  举报

导航