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

更新于: 2020-04-30

407 次查看

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告