题目描述
请实现有重复数字的有序数组的二分查找。
输出在数组中第一个大于等于查找值的位置,如果数组中不存在这样的数,则输出数组长度加一。
示例
输入
5,4,[1,2,4,4,5]
输出
3
思路描述
- 首先判断数组中是否存在该数,若不存在,直接输出数组长度加一
- 由于输出大于等于查找值的位置,我们的left和right是能相同的,故可设定我们的位置pos值是能够取到mid的值的
代码描述:
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
if(a[n-1]<v) return n+1;
int left=0,right=n-1,mid,pos;
while(left<=right)
{
mid = (left+right)/2;
if(a[mid]>=v)
{
right = mid-1;
pos = mid;
}
if(a[mid]<v)
left = mid+1;
}
return pos+1;
}
};