【Leetcode】1060. Missing Element in Sorted Array

题目地址：

https://leetcode.com/problems/missing-element-in-sorted-array/

给定一个升序数组$A$，其中数字两两不同。问下面的序列$S=(A[0]+1,A[0]+2,...)$中的数字，第$k$个不在数组中的数是几。

思路是二分。要找到第$k$个不在数组中的数是几，我们先定义$A[i]$左边miss的数的个数是$f(i)$，则： f ( i ) = # { S ∣ s < A [ i ] , s ∈ S − { A [ 0 ] , A [ 1 ] , . . . , A [ i − 1 ] } } f(i)=\#\{S|s<A[i],s\in S-\{A[0],A[1],...,A[i-1]\}\} f(i)=#{S∣s<A[i],s∈S−{A[0],A[1],...,A[i−1]}}举个例子，设$A=(4,7,9,10)$，那么$9$左边miss的数是$5,6,8$，也就是$f(2)=3$。显然$i$越大，左边miss的数也就越多。我们先找最大的$i$使得$f(i)\le k-1$，这样第$k$个miss的数就是$$k-f(i)+A[i]$$这个式子的意思是，$A[i]$左边已经miss掉$f(i)$个数，把这$f(i)$个数扣除掉之后，要找的数就是比$A[i]$大的第$k-f(i)$个miss的数，那就加上去就行了。代码如下：

public class Solution {
    public int missingElement(int[] nums, int k) {
        // 找左边miss数小于等于k的最大的数
        int l = 0, r = nums.length - 1;
        while (l < r) {
            int m = l + (r - l + 1 >> 1);
            if (missCount(nums, m) <= k - 1) {
                l = m;
            } else {
                r = m - 1;
            }
        }
        
        return nums[l] + k - missCount(nums, l);
    }
    
    // 返回nums[i]左边的miss的数的个数
    private int missCount(int[] nums, int i) {
        return nums[i] - nums[0] - i;
    }
}

时间复杂度$O(\log n)$，空间$O(1)$。