LeetCode 1539. Kth Missing Positive Number

原题链接在这里:https://leetcode.com/problems/kth-missing-positive-number/description/

题目:

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

Return 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

Follow up:

Could you solve this problem in less than O(n) complexity?

题解:

Assume x is the final result, from 1 to x, there are m numbers not missing.

Try to find the m position.

Use the binary search, arr[mid] - mid - 1 is the missing numbers up to index mid.

if it is smaller than k, than m must be on the right, l = mid + 1. 

else, than m could on the left including current m. r = m.

Note: initialize r as arr.length.

Time Complexity: O(logn). n = arr.length.

Space: O(1).

AC Java:

 1 class Solution {
 2     public int findKthPositive(int[] arr, int k) {
 3         int l = 0;
 4         int r = arr.length;
 5         while(l < r){
 6             int mid = l + (r - l) / 2;
 7             if(arr[mid] - mid  - 1 < k){
 8                 l = mid + 1;
 9             }else{
10                 r = mid;
11             }
12         }
13 
14         return l + k;
15     }
16 }

类似Missing Element in Sorted Array.

posted @ 2024-04-05 10:33  Dylan_Java_NYC  阅读(9)  评论(0编辑  收藏  举报