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

1. 题目

给出一个有序数组 A,数组中的每个数字都是 独一无二的,找出从数组最左边开始的第 K 个缺失数字。

示例 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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/missing-element-in-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

2.1 一次遍历

  • 相邻的数做差,进行判断,对 k 进行更新,直到 k <= 0 停止
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;
    }
};

124 ms 29.6 MB

2.2 二分查找

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;
    }
};

144 ms 29.6 MB


我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!
Michael阿明

猜你喜欢

转载自blog.csdn.net/qq_21201267/article/details/107538389
今日推荐