二分查找，lower_bound，upper_bound

LeetCode 15. 3Sum

想到的算法是排序之后二重循环a和b，然后二分查找第三个c，时间复杂度 $O(n^2log(n))$ 。于是想手写一下二分查找。搜到了LeetCode: Search for a Range这一题。于是就写了lower_bound和upper_bound。

lower_bound & upper_bound

参考定义
* ForwardIter lower_bound(ForwardIter first, ForwardIter last,const _Tp& val)算法返回一个非递减序列[first, last)中的第一个大于等于值val的位置。
* ForwardIter upper_bound(ForwardIter first, ForwardIter last, const _Tp& val)算法返回一个非递减序列[first, last)中第一个大于val的位置。

Code

针对Search for a Range 这一题的实现如下。但是最后返回的结果还需要再经过判断range是否存在来决定。

int lower_bound(vector<int>& nums, int target) {
    // [lo, hi)
    int lo = 0;
    int hi = nums.size();
    int pos = -1;
    while (lo < hi) {
        int mi = lo + (hi - lo) / 2;
        target > nums[mi] ?  lo = mi + 1, pos = lo : hi = mi, pos = hi ;
    }
    return pos;
}

int upper_bound(vector<int>& nums, int target) {
    // [lo, hi)
    int lo = 0; 
    int hi = nums.size();
    int pos = -1;
    while (lo < hi) {
        int mi = lo + (hi - lo) / 2;
        target < nums[mi] ? hi = mi, pos = hi : lo = mi + 1, pos = lo ;
    }
    return pos;
}