本文共 1130 字,大约阅读时间需要 3 分钟。
题目:
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 = 5Output: 9Explanation: 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 = 2Output: 6Explanation: 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; } };转载地址:http://saqo.baihongyu.com/