题目:
请实现有重复数字的升序数组的二分查找。
输出在数组中第一个大于等于查找值的位置,如果数组中不存在这样的数,则输出数组长度加一。
本质上还是二分查找,只是在对mid值和查找值index之间的处理有些不同。
采用递归实现的二分查找的话,在进行index和arr[mid]的判断的时候分成了三种情况,等于,大于和小于,等于的时候直接返回index,大于的时候去前半段进行查找,小于的时候去后半段查找,但是题目的要求是查找“第一个大于等于index”的位置,没有的话输出长度加一,并且数组中是包含重复元素的,因此对于大于和等于的情况,需要先记录下mid+1作为中间结果,然后去半段查找,找到其他符合的去更新res,小于的话就直接去后半段找,不做操作。
完整代码如下:
public static int upper_bound_ (int n, int v, int[] a) {
// write code here
return binarySearch(0, n-1, v, a,n+1);
}
public static int binarySearch(int begin,int end,int index,int[] a,int res) {
if(begin>end) {
return res;
}
int mid = (begin+end)/2;
if(a[mid]>=index) {
res = mid+1;
return binarySearch(begin, mid-1, index, a,res);
}else {
return binarySearch(mid+1, end, index, a,res);
}
}