1539. Kth Missing Positive Number

题目：

Given an array arr of positive integers sorted in a strictly increasing order, and an integer k.

Find the kth positive integer that is missing from this array.

 

Example 1:

Input: arr = [2,3,4,7,11], k = 5
Output: 9
Explanation: The missing positive integers are [1,5,6,8,9,10,12,13,...]. The 5th missing positive integer is 9.

Example 2:

Input: arr = [1,2,3,4], k = 2
Output: 6
Explanation: The missing positive integers are [5,6,7,...]. The 2nd missing positive integer is 6.

 

Constraints:

• 1 <= arr.length <= 1000
• 1 <= arr[i] <= 1000
• 1 <= k <= 1000
• arr[i] < arr[j] for 1 <= i < j <= arr.length

 

 

 

思路：

我们需要两个int，一个是count来track目前为止已经缺失了多少个数，另一个是cur，在不缺失的情况下，当前的数字应该是多少。然后遍历数组，因为cur初始化是0，而且正常情况的数组也是应该从1开始，因此循环开始cur先加1，然后如果cur和当前的值不相等，则while循环增加count和cur，在这过程中需要注意如果count等于k了，则直接返回cur即可，即不一定需要遍历完全部数组。最后一个注意点是k可能超过数组，比如[1,2,3,4]，k=2，则缺失了5和6，这部分尾部缺失我们也需要进行确认并输出。

 

 

 

代码：

class Solution {
public:
    int findKthPositive(vector<int>& arr, int k) {
        int count = 0;
        int cur = 0;
        for (int i = 0; i < arr.size(); i++)
        {
            cur++;
            while (cur < arr[i])
            {
                count++;
                if (count == k)
                    return cur;
                cur++;
            }
        }
        while (count < k)
        {
            cur++;
            count++;
        }
        return cur;
    }
};