请实现有重复数字的有序数组的二分查找。
输出在数组中第一个大于等于查找值的位置,如果数组中不存在这样的数,则输出数组长度加一。
二分答案模板题,区间中满足条件的最小值。
class Solution {
public:
/**
* 二分查找
* @param n int整型 数组长度
* @param v int整型 查找值
* @param a int整型vector 有序数组
* @return int整型
*/
int upper_bound_(int n, int v, vector<int>& a) {
// write code here
int l = 0;
int r = n - 1;
int ans = -1;
while (l <= r) {
int mid = (l + r) >> 1;
if (a[mid] >= v) {
ans = mid;
r = mid - 1;
} else {
l = mid + 1;
}
}
if (ans == -1) {
return n + 1;
} else {
return ans + 1;
}
}
};