给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
这道题目的前提是数组为有序数组,同时题目还强调数组中无重复元素,因为一旦有重复元素,使用二分查找法返回的元素下标可能不是唯一的,这些都是使用二分法的前提条件,当大家看到题目描述满足如上条件的时候,可要想一想是不是可以用二分法了。
示例 1:
输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4
示例 2:
输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1
非递归写法:
public int binarySearch(int[] nums, int target) {
// 避免当 target 小于nums[0] nums[nums.length - 1]时多次循环运算
if (target < nums[0] || target > nums[nums.length - 1]) {
return -1;
}
int left = 0, right = nums.length - 1;
while (left <= right) {
int mid = left + ((right - left) >> 1);
if (nums[mid] == target)
return mid;
else if (nums[mid] < target)
left = mid + 1;
else if (nums[mid] > target)
right = mid - 1;
}
return -1;
}
二分查找的关键是要定义左右边界,规定好区间,由于给定的数组为有序数组,即从小到大排列,我们就通过 left+( (right - left ) / 2)或者( left+right ) / 2来获得中间值的下标(由于后者会有超出数值范围的风险,通常我们使用前者)
当我们获取到数组的中间值大于我们要查找的值时,说明我们想要查找的值在中间值的左边,即我们想要查找的值不在中间和数组的右边,那么就将右边的指针指向mid-1,对新的数组再继续进行判断,反之亦然。
>>1是指将数值类型的二进制位整体向右移动一位,例如0000 0110 >> 1 = 0000 0011,前者转换为十进制的结果为6,后者转换为十进制的结果为3,等同于除以2的结果,由于JVM进行了算法的优化,所以两种方法都可以使用。
递归写法:
public static int binarySearch(int[] nums,int target, int left, int right) {
if (left > right) {
return -1;
}
int mid = left+((right-left) >> 1);
if (nums[mid] == target)
return mid;
else if (nums[mid] < target)
return binarySearch(nums, target, mid + 1, right);
else
return binarySearch(nums, target, left, mid - 1);
}
总结
二分法是非常重要的基础算法,为什么很多同学对于二分法都是一看就会,一写就废?
其实主要就是对区间的定义没有理解清楚,在循环中没有始终坚持根据查找区间的定义来做边界处理。
区间的定义就是不变量,那么在循环中坚持根据查找区间的定义来做边界处理,就是循环不变量规则。
相信看完本篇应该对二分法有更深刻的理解了。