二分查找

题目链接

Solution

查找数组中第一个大于等于查找值的位置,显然可以二分法做。
因为数组是有序的,所以选取中间的位置,然后判断是否比要找的数大。如果中间的数大于这个数,显然答案在左边,否则在右边。
每次将区间长度缩小一半,故复杂度是

Code

class Solution {
public:
    int upper_bound_(int n, int v, vector<int>& a) {
        int L = 0, R = n - 1, ans = n;
        while (L <= R) {
            int mid = (L + R) / 2;
            if (a[mid] >= v) ans = mid, R = mid - 1;
            else L = mid + 1;
        }
        return ans + 1;
    }
};