简单化写法解决二分

最新推荐文章于 2025-09-13 21:00:42 发布

原创最新推荐文章于 2025-09-13 21:00:42 发布 · 101 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

31 篇文章

订阅专栏

博客内容展示了如何使用二分查找算法找到不小于目标值的最左边的数字，以及等于目标值的最右边的数字。通过两种不同实现方式，讨论了左闭右开和左闭右闭边界条件的影响，并提供了与标准库`lower_bound`函数的比较。

简单化写法解决二分

找到不小于某个值的最左边的数字

#include <algorithm>
#include <iostream>
#include <vector>

using namespace std;

int bfind(
    vector<int>& data,
    int target) {  // 函数是在data中找到第一个不小于target的数，相当于lower_bound

  int l = 0;
  int r = data.size() - 1;
  int res = -1;
  // 所以这里维持着一个左闭右闭

  while (l <= r) {
    int mid = l + (r - l) / 2;
    if (data[mid] < target) {
      l = mid + 1;
    } else if (data[mid] >= target) {
      res = mid;
      r = mid - 1;
    }
  }
  return res;
}

int main() {
  vector<int> data = {5, 8, 8, 8, 8, 9, 13};

  cout << bfind(data, 8) << endl;

  auto iter = lower_bound(data.begin(), data.end(), 8);
  if (iter == data.end()) {
    cout << "not find" << endl;
    return 0;
  }
  cout << std::distance(data.begin(), iter) << endl;

  cout << *iter << endl;
}

#include <algorithm>
#include <iostream>
#include <vector>

using namespace std;

int bfind(
    vector<int>& data,
    int target) {  // 函数是在data中找到第一个不小于target的数，相当于lower_bound

  int l = 0;
  int r = data.size();
  int res = -1;
  // 所以这里维持着一个左闭右开

  while (l < r) {
    int mid = l + (r - l) / 2;
    if (data[mid] < target) {
      l = mid + 1;
    } else if (data[mid] >= target) {
      res = mid;
      r = mid;
    }
  }
  return res;
}

int main() {
  vector<int> data = {5, 8, 8, 8, 8, 9, 13};

  cout << bfind(data, 8) << endl;

  auto iter = lower_bound(data.begin(), data.end(), 8);
  if (iter == data.end()) {
    cout << "not find" << endl;
    return 0;
  }
  cout << std::distance(data.begin(), iter) << endl;

  cout << *iter << endl;
}

找到等于某个值的最右边的数字

int bfind(
    vector<int>& data,
    int target) {  

  int l = 0;
  int r = data.size();
  int res = -1;
  // 所以这里维持着一个左闭右开

  while (l < r) {
    int mid = l + (r - l) / 2;
    if (data[mid] < target) {
      l = mid + 1;
    } else if (data[mid] == target) {
      res = mid;
      l = mid + 1;
    }
    if (data[mid] > target) {
      r = mid;
    }
  }
  return res;
}

简单点直接加一个res,这个res的好处在于能够返回没有找到的情况，如果直接返回r或者l是不能处理没找到的情况

int bfind(
    vector<int>& data,
    int target) { 

  int l = 0;
  int r = data.size()-1;
  int res = -1;
  // 所以这里维持着一个左闭右闭

  while (l <= r) {
    int mid = l + (r - l) / 2;
    if (data[mid] < target) {
      l = mid + 1;
    } else if (data[mid] == target) {
      res = mid;
      l = mid + 1;
    }
    if (data[mid] > target) {
      r = mid-1;
    }
  }
  return res;
}