leetcode1539 第k个缺失的正整数

思路1：

暴力枚举。

实现1：

class Solution
{
public:
    int findKthPositive(vector<int>& arr, int k)
    {
        int n = arr.size();
        vector<int> cnt(2001, 0);
        for (int i = 0; i < n; i++) cnt[arr[i]] = 1;
        int x = 0, res = 0;
        for (int i = 1; i <= 2000; i++)
        {
            if (!cnt[i]) x++;
            if (x == k) { res = i; break; }
        }
        return res;
    }
};

思路2:

根据数组中元素的数值大小和所在位置的相对关系计算答案。

实现2:

class Solution
{
public:
    int findKthPositive(vector<int>& arr, int k)
    {
        if (arr[0] > k) return k;
        int cnt = arr[0] - 1, n = arr.size();
        for (int i = 0; i < n - 1; i++)
        {
            int d = arr[i + 1] - arr[i] - 1;
            if (cnt + d >= k) return arr[i] + k - cnt;
            cnt += d;
        }
        return arr[n - 1] + k - cnt;
    }
};

思路3:

在思路2的基础上更进一步，使用二分查找。

实现3:

class Solution
{
public:
    int cnt(vector<int>& arr, int m)
    {
        return arr[m] - m - 1;
    }
    int findKthPositive(vector<int>& arr, int k)
    {
        int n = arr.size();
        int tmp = arr[n - 1] - n;
        if (tmp < k) return arr[n - 1] + k - tmp;
        int l = 0, r = n - 1, p = -1;
        while (l <= r)
        {
            int m = l + r >> 1;
            if (arr[m] - m - 1 < k) l = m + 1;
            else
            {
                p = m; r = m - 1;
            }
        }
        tmp = arr[p] - p - 1;
        return arr[p] + k - tmp - 1;
    }
};