数组中缺失的最小正整数

华电北风吹
日期：2016/2/24

题目链接：
https://leetcode.com/problems/first-missing-positive/

题目描述：
对一个未排序的数组，找出缺失的最小正整数。
例如，对于数组[1,2,0]返回3；对于数组[3,4,-1,1]返回2。
算法时间复杂度应该为O(n)和常数的空间复杂度。

解析：
一、如果数组重元素没有重复
借助于快排把数据划分在给定元素左右，这里选择输入数组的中间元素递归划分。比如如果输入的数组在low-high区间内，将数据按照中间元素(low+high)/2划分，然后看中间元素值是否为(low+high)/2来判断下次递归是左半部分还是右半部份。最后的时间复杂度为3n=n+n+n/2+n/4…

#include <iostream>
#include <vector>
using namespace std;

class Solution
{
public:
    int FindLeastInteger(vector<int> input)
    {
        int k = -1;
        for (int i = 0; i < input.size(); i++)
        {
            if (input[i]>0)
            {
                k++;
                input[k] = input[i];
            }
        }
        return DivisionFunction(input, 0, k);
    }

    int DivisionFunction(vector<int>input, int low, int high)
    {
        if (low == high)
        {
            if (input[low] == low + 1)
                return low + 2;
            else
                return low + 1;
        }
        int mid = (low + high) / 2 + 1;
        int k = low;
        for (int i = low; i <= high; i++)
        {
            if (input[i] <= mid)
            {
                int temp = input[k];
                input[k] = input[i];
                input[i] = temp;
                k++;
            }
        }
        if (k == low)
            return low + 1;
        if (k == mid)
            return DivisionFunction(input, mid, high);
        else
            return DivisionFunction(input, low, k - 1);
    }
};

int main(int argc, _TCHAR* argv[])
{
    vector<int> v{ 5, 3, 4, 2, -4, 6, 8, 9 ,1 };
    Solution s;
    cout << s.FindLeastInteger(v);

    cout << endl;
    getchar();
    return 0;
}

二、如果数组中元素包含重复元素：
借助于计数排序的思路，对数组进行排序。然后找出最小的缺失元素。

#include <iostream>
#include <vector>
using namespace std;

class Solution
{
public:
    int firstMissingPositive(vector<int> nums)
    {
        int n = nums.size(), i = 0;
        while (i < n)
        {
            if (nums[i] > 0 && nums[i] <= nums.size() && nums[nums[i] - 1] != nums[i])
                swap(nums, i, nums[i] - 1);
            else
                i++;
        }
        i = 0;
        while (i < n && nums[i] == i + 1)
        {
            i++;
        }
        return i + 1;
    }

    void swap(vector<int> &nums, int i, int j)
    {
        nums[i] = nums[i] ^ nums[j];
        nums[j] = nums[i] ^ nums[j];
        nums[i] = nums[i] ^ nums[j];
    }
};

int main(int argc, _TCHAR* argv[])
{
    vector<int> v{ -1, 4, 2, 1, 9, 10 };
    Solution s;
    cout << s.firstMissingPositive(v) << endl;
    getchar();
    return 0;
}