lower_bound和upper_bound的实现

Alvin's Tech Blog

已于 2024-06-26 21:18:22 修改

阅读量627

点赞数

分类专栏：算法题目 C/C++学习文章标签：排序算法算法

于 2022-02-13 00:12:03 首次发布

本文链接：https://blog.csdn.net/pan_chengyuan/article/details/122904312

版权

算法题目同时被 2 个专栏收录

4 篇文章 0 订阅

订阅专栏

C/C++学习

3 篇文章 0 订阅

订阅专栏

本文详细介绍了如何实现二分查找算法的两个变种：lower_bound和upper_bound。这两个函数分别用于在有序数组中找到大于等于目标值的第一个元素和大于目标值的第一个元素的索引。博客通过代码示例展示了它们的工作原理，强调了在实现过程中对细节的把握对结果的重要性。

摘要由CSDN通过智能技术生成

binary search算法特别讲究细节，微小的改动都会带来不同的结果，这篇博客将放上lower_bound和upper_bound的实现

lower_bound的实现

说明：inputs是升序排序好的数组，target是要搜索的元素

// Function to perform lower bound binary search
int lowerBound(const std::vector<int>& data, int target) {
    int low = 0;
    int high = data.size(); // Note: high is set to data.size() not data.size() - 1

    while (low < high) {
        int mid = low + (high - low) / 2;
        if (data[mid] < target) {
            low = mid + 1;
        } else {
            high = mid;
        }
    }

    return low; // Returns the index of the first element >= target
}

upper_bound的实现

// Function to perform upper bound binary search
int upperBound(const std::vector<int>& data, int target) {
    int low = 0;
    int high = data.size(); // Note: high is set to data.size(), not data.size() - 1

    while (low < high) {
        int mid = low + (high - low) / 2;
        if (data[mid] <= target) {
            low = mid + 1; // Move low up to exclude the current mid
        } else {
            high = mid;
        }
    }

    return low; // Returns the index of the first element > target
}