C++ 中排序数组中的缺失元素
假设我们有一个包含唯一数字的排序数组 A,我们需要找到从数组最左侧数字开始的第 K 个缺失数字。例如,如果数组为 [4,7,9,10],并且 k = 1,则该元素将为 5。
为了解决这个问题,我们将遵循以下步骤:
n := 数组的大小,设置 low := 0 和 high := n – 1
如果 nums[n - 1] – nums[0] + 1 – n < k,则
返回 nums[n - 1] + (k – (nums[n - 1] – nums[0] + 1 – n))
当 low < high – 1
mid := low + (high - low)/2
present := mid – low + 1
absent := nums[mid] – nums[low] + 1 – present
如果 absent >= k,则 high := mid,否则将 k 减去 absent,low := mid
返回 nums[low] + k
让我们看看下面的实现来更好地理解:
示例
#include <bits/stdc++.h> using namespace std; class Solution { public: int missingElement(vector<int>& nums, int k) { int n = nums.size(); int low = 0; int high = n - 1; if(nums[n - 1] - nums[0] + 1 - n < k) return nums[n - 1] + (k - ( nums[n - 1] - nums[0] + 1 - n)) ; while(low < high - 1){ int mid = low + (high - low) / 2; int present = mid - low + 1; int absent = nums[mid] - nums[low] + 1 - present; if(absent >= k){ high = mid; }else{ k -= absent; low = mid; } } return nums[low] + k; } }; main(){ vector<int> v = {4,7,9,10}; Solution ob; cout << (ob.missingElement(v, 1)); }
输入
[4,7,9,10] 1
输出
5
广告