LeetCode 1060. 有序数组中的缺失元素（二分查找）

示例 1：
输入：A = [4,7,9,10], K = 1
输出：5
解释：
第一个缺失数字为 5 。

示例 2：
输入：A = [4,7,9,10], K = 3
输出：8
解释： 
缺失数字有 [5,6,8,...]，因此第三个缺失数字为 8 。

示例 3：
输入：A = [1,2,4], K = 3
输出：6
解释：
缺失数字有 [3,5,6,7,...]，因此第三个缺失数字为 6 。
 
提示：
1 <= A.length <= 50000
1 <= A[i] <= 1e7
1 <= K <= 1e8

class Solution {
public:
    int missingElement(vector<int>& nums, int k) {
    	int i = 0, n = nums.size(), ans;
    	for(i = 0; i < n-1; ++i)
    	{
    		if(nums[i+1]-nums[i]-1 < k)
    			k -= nums[i+1]-nums[i]-1;
    		else
    		{
    			ans = nums[i]+k;
    			return ans;
    		}
    	}
    	return nums[n-1]+k;
    }
};

class Solution {
public:
    int missingElement(vector<int>& nums, int k) {
        int l = 0, r = nums.size()-1, mid, miss;
        while(l <= r)
        {
            mid = l+((r-l)>>1);
            miss = countmissing(nums, mid);
            if(miss < k)
                l = mid+1;
            else if(miss > k)
                r = mid-1;
            else
                return nums[mid]-1;

        }
        return nums[r]+k-countmissing(nums,r);
    }
    int countmissing(vector<int>& nums, int i)
    {
        return nums[i]-nums[0]-i;
    }
};