1539. 第 k 个缺失的正整数

1539. 第 k 个缺失的正整数https://leetcode.cn/problems/kth-missing-positive-number/

难度简单106

给你一个 严格升序排列 的正整数数组 arr 和一个整数 k 。

请你找到这个数组里第 k 个缺失的正整数。

示例 1：

输入：arr = [2,3,4,7,11], k = 5
输出：9
解释：缺失的正整数包括 [1,5,6,8,9,10,12,13,...] 。第 5 个缺失的正整数为 9 。

示例 2：

输入：arr = [1,2,3,4], k = 2
输出：6
解释：缺失的正整数包括 [5,6,7,...] 。第 2 个缺失的正整数为 6 。

提示：

• 1 <= arr.length <= 1000
• 1 <= arr[i] <= 1000
• 1 <= k <= 1000
• 对于所有 1 <= i < j <= arr.length 的 i 和 j 满足 arr[i] < arr[j] 

通过次数25,597提交次数47,444

class Solution {
    public int findKthPositive(int[] arr, int k) {
       
        // 缺失数在数组外的
        int n = arr.length-1;
        if((arr[n]-n-1)<k) return k-(arr[n]-n-1)+arr[n];

        if(arr[0]>k) return k;

        //缺失数在数组内的：二分查找
        int start = 0;
        int end = arr.length-1;
        int mid;
        while(start<=end)
        {
            mid = (end-start)/2+start;
            if(arr[mid]-mid-1==k && arr[mid+1]-mid-2>=k)
            {
                System.out.println(mid+"  "+arr[mid]);
                int num = arr[mid]-mid-1-k;
                System.out.println("num="+num);
                int x = arr[mid]-1;
                while(num>=0)
                {
                    if(x==arr[mid-1])
                    {
                        x--;
                    }
                    else num--;
                }
                return x;
            }
            if(arr[mid]-mid-1<k && arr[mid+1]-mid-2>=k)
            {
                System.out.println(mid+1+"  "+arr[mid+1]);
                return arr[mid+1]-(arr[mid+1]-mid-2-k+1);
            }

            if(arr[mid]-mid-1>k) end = mid-1;
            else start=mid+1;
        }
        return 0;

    }
}