二分查找算法是在一个数组中查询一个指定数值下标的算法方案。二分查找算法可以让数据的查询变得高效,它在很多领域及技术的底层原理中都被广泛使用到,例如 MySQL 的索引实现、JDK 中的源码实现等。
需要注意的是,并不是任何数组都满足使用二分查找算法进行数据查询的条件,二分查找算法的使用需要满足一定的前提条件。对于被二分查找的数组而言,数组必须是已经排好序的数组,但是可以忽略是正向从小到大排序还是逆向从大到小排序,因为不论是正向排序还是逆向排序,都在二分查找算法的适用范围内。
二分查找算法的实现原理
- 定义左边界下标与右边界下标以确定搜索范围;
- 循环执行二分查找进行数据查找:
- 获取中间下标及该下标对应的数值;
- 将该数值与需要查找的数值进行比较:
- 若该数值与需要查找的数值相同,则直接返回下标;
- 若该数值小于需要查找的数值,说明需要查找的数值数据在右,则将左边界下标定义为
中间下标+1
并重新查找; - 若该数值大于需要查找的数值,说明需要查找的数值数据在左,则将右边界下标定义为
中间下标-1
并重新查找;
- 当左下标大于右下标时结束循环(此时说明需要查找的数据在数组中并不存在);
二分查找算法的实现案例
以下是一个通过 Java 实现二分查找的方法案例:
public static int binarySearch(int[] nums, int num) {
int left = 0;
int right = nums.length - 1;
while (left <= right) {
int middle = (left + right) >>> 1;
if (nums[middle] == num) {
return middle;
}
if (nums[middle] < num) {
left = middle + 1;
}
if (nums[middle] > num) {
right = middle - 1;
}
}
return -1;
}
上述案例中,通过
(left + right) >> 1
计算middle
值,主要是为了避免整数溢出问题。在 JDK 中的 Arrays.binarySearch 也是通过>>>
解决整数溢出问题。
定义一个 main
方法对上述方法案例进行测试:
public static void main(String[] args) {
int[] nums = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
for (int num : nums) {
int findNumIndex = binarySearch(nums, num);
System.out.println("查找元素" + num + "的下标为:" + findNumIndex);
}
}
main
方法的执行结果为:
查找元素1的下标为:0
查找元素2的下标为:1
查找元素3的下标为:2
查找元素4的下标为:3
查找元素5的下标为:4
查找元素6的下标为:5
查找元素7的下标为:6
查找元素8的下标为:7
查找元素9的下标为:8
查找元素10的下标为:9