解决方案
在数组中搜索数据,一般我们都是采用遍历的方式,复杂度为O(n)。有序数组中包含大小信息,可以将其视作二叉搜索树结构,故该问题可以使用二分法解决。关于二分法的一些介绍,可以提供一个传送门:二分法。
算法实现
一般而言,二分法都有两个版本,一种是C++中的左闭右开(下界)版本,一种是Java中的左闭右闭版本。
左闭右开版本
左闭右开版本的代码如下所示:
public int binarySearch(int[] array, int target){
int low = 0, high = array.length;
while(low < high){
int mid = (low + high) >>> 1;
if(array[mid] < target){
low = mid + 1;
}else{
high = mid;
}
}
return low;
}
该算法需要重点关注返回的索引值,如果数组中存在该值,则返回数组中存在该数组第一次出现的索引值;如果数组中不存在该值,则返回数组中第一个比它大的索引值,可以能出现超出索引范围的情况。
左闭右闭版本
左闭右闭版本的代码如下所示:
public int binarySearch(int[] array, int target){
int low = 0, high = array.length - 1;
while(low <= high){
int mid = (low + high) >>> 1, midVal = array[mid];
if(midVal < target){
low = mid + 1;
}else if(midVal > target){
high = mid - 1;
}else{
return mid;
}
}
return -(low + 1);
}
与前面索引返回值不同,如果数组中存在该值,则返回第一次搜索到的索引值,这里的索引值并不是首个值的索引值,位置不确定;如果数组中不存在该值,则返回-(low + 1),其值为负,low表示左搜索索引值,它的位置一般都是比搜索值大的首索引值,所以可能出现超出索引范围的情况。根据位运算 -x = ~(x + 1),即x = ~(-(x + 1)),因此我们可以对返回值取反来获取low值。
应用场景
1、当我们需要判断数组中存在某个值时,一般使用左闭右闭版本的二分法,直接通过返回值是否为非负数来进行判断。
2、当我们需要找到数组中某个值的位置时,如果不强调是首位置,两个版本的二分法皆可以;如果强调是首位置,则需要使用左闭右开版本的二分法。
3、当我们需要找到大于等于某个值位置时,优先选择左闭右开版本,左闭右闭版本也行,只是多一个取反操作。
问题小结
当我们遇到有序数组问题时,一定要想到二分法,有一些问题不是简单给你有序数组,比如判断某个数据在什么范围内,虽然没直接说是有序数组,但是我们可以使用该思想进行解决,多思考有序环境,不能仅仅局限在有序数组。