牛客题霸--二分查找题解

二分查找

题目链接

Solution

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

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;
    }
};