二分查找，lower_bound，upper_bound

LeetCode 15. 3Sum

想到的算法是排序之后二重循环a和b，然后二分查找第三个c，时间复杂度 O(n2log(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;}