二分查找模板
int binarySearch(int[] nums, int target) {
int left = 0, right = ...;
while(...) {
int mid = left + (right - left) / 2;
if (nums[mid] == target) {
...
} else if (nums[mid] < target) {
left = ...
} else if (nums[mid] > target) {
right = ...
}
}
return ...;
}
分析二分查找的一个技巧是:不要出现 else,而是把所有情况用 else if 写清楚,这样可以清楚地展现所有细节。
其中...
标记的部分,就是可能出现细节问题的地方,当你见到一个二分查找的代码时,首先注意这几个地方。后文用实例分析这些地方能有什么样的变化。
另外声明一下,计算 mid 时需要防止溢出,代码中left + (right - left) / 2
就和(left + right) / 2
的结果相同,但是有效防止了left
和right
太大直接相加导致溢出。
当left取值为0,而right取值num.length-1()的话,while(left<=right);而right取值为num.length()的话,while(left<right)
-
基本二分搜索
int binarySearch(int[] nums, int target) { int left = 0; int right = nums.length - 1; // 注意 while(left <= right) { int mid = left + (right - left) / 2; if(nums[mid] == target) return mid; else if (nums[mid] < target) left = mid + 1; // 注意 else if (nums[mid] > target) right = mid - 1; // 注意 } return -1; }
-
寻找左侧边界的二分搜索–当nums[mid]==target,right = mid - 1;return left,加left超出数组长度判断
int left_bound(int[] nums, int target) { int left = 0, right = nums.length - 1; // 搜索区间为 [left, right] while (left <= right) { int mid = left + (right - left) / 2; if (nums[mid] < target) { // 搜索区间变为 [mid+1, right] left = mid + 1; } else if (nums[mid] > target) { // 搜索区间变为 [left, mid-1] right = mid - 1; } else if (nums[mid] == target) { // 收缩右侧边界 right = mid - 1;//倒数第二步是l=r=mid,若nums[mid]==target,r=mid-1,r<l,跳出循环,返回l=mid=target。!!注意正因为这里当nums[mid] == target,right=mid-1,mid=right+1,又因为while循环终止的条件是left>right,所以最终返回的是left } } // 检查出界情况,如下图 if (left >= nums.length || nums[left] != target) return -1; return left;//!! }
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-9Ubn6yhI-1637549881664)(C:\Users\WXY\AppData\Roaming\Typora\typora-user-images\image-20210822110203655.png)]
4、为什么该算法能够搜索左侧边界?
答:关键在于对于
nums[mid] == target
这种情况的处理:if (nums[mid] == target) right = mid-1;
可见,找到 target 时不要立即返回,而是缩小「搜索区间」的上界
right
,在区间[left, mid]
中继续搜索,即不断向左收缩,达到锁定左侧边界的目的。 -
寻找右侧边界的二分搜索–当nums[mid]==target,left = mid + 1; return right,加right下标小于数组最左侧下标判断
int right_bound(int[] nums, int target) { int left = 0, right = nums.length - 1; while (left <= right) { int mid = left + (right - left) / 2; if (nums[mid] < target) { left = mid + 1; } else if (nums[mid] > target) { right = mid - 1; } else if (nums[mid] == target) { // 这里改成收缩左侧边界即可 left = mid + 1;//mid=left-1 } } // 这里改为检查 right 越界的情况,见下图 if (right < 0 || nums[right] != target) return -1; return right;//while结束循环的条件是left>right,mid=left-1,right=mid=target }
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-MZP2Ih6e-1637549881665)(C:\Users\WXY\AppData\Roaming\Typora\typora-user-images\image-20210822110130811.png)]
1、分析二分查找代码时,不要出现 else,全部展开成 else if 方便理解。
2、注意「搜索区间」和 while 的终止条件,如果存在漏掉的元素,记得在最后检查。