二分搜索
二分搜索介绍
二分查找是一种在每次比较之后将查找空间一分为二的算法。二分查找维护查找空间的左、右和中间指示符,并比较查找目标,如果条件不满足或值不相等,则清除目标不可能存在的那一半,并在剩下的一半上继续查找,直到成功为止。如果查找以空的一半结束,则无法满足条件,并且无法找到目标。
四个要素:
- 初始化:start=0、end=len-1
- 循环退出条件:start + 1 < end
- 比较中点和目标值:A[mid] ==、 <、> target
- 判断最后两个元素是否符合:A[start]、A[end] ? target
识别二分搜索
- 每次需要查找集合中的索引或元素时,都应该考虑二分查找。
- 集合有序(如果集合是无序的,我们可以在应用二分查找之前先对其进行排序)。
实现二分搜索的方法
详情点击:二分搜索模板
方法一:
- 查找条件可以在不与元素的两侧进行比较的情况下确定(或使用它周围的特定元素)。
- 不需要后处理,因为每一步中,你都在检查是否找到了元素。如果到达末尾,则知道未找到该元素。
int binarySearch(int[] nums, int target){
if(nums == null || nums.length == 0)
return -1;
int left = 0, right = nums.length - 1;
while(left <= right){
// Prevent (left + right) overflow
int mid = left + (right - left) / 2;
if(nums[mid] == target){ return mid; }
else if(nums[mid] < target) { left = mid + 1; }
else { right = mid - 1; }
}
// End Condition: left > right
// 应该是right略小于target,left略大于target
return -1;
}
跳出时,变量 i 和 j 分别指向 “右子数组的首位元素” 和 “左子数组的末位元素” 。因此返回 i 即可。
方法二:
- 搜索条件需要访问元素的直接左右邻居。
- 使用元素的邻居来确定它是向右还是向左。
- 保证查找空间在每个步骤中至少有 3 个元素。
- 需要进行后处理。 当剩下 2 个元素时,循环 / 递归结束。 需要评估其余元素是否符合条件。
int binarySearch(int[] nums, int target) {
if (nums == null || nums.length == 0)
return -1;
int left = 0, right = nums.length - 1;
while (left + 1 < right){
// Prevent (left + right) overflow
int mid = left + (right - left) / 2;
if (nums[mid] == target) {
return mid;
} else if (nums[mid] < target) {
left = mid;
} else {
right = mid;
}
}
// Post-processing:
// End Condition: left + 1 == right
if(nums[left] == target) return left;
if(nums[right] == target) return right;
return -1;
}
总结:
- 大部分场景方法二都能解决问题,而且还能找第一次/最后一次出现的位置
- 如果是最简单的二分搜索,不需要找第一个、最后一个位置、或者是没有重复元素,可以使用方法一
题目一
主站-35. 搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
解题思路
这是最简单的二分查找,最后要找的目标值不用与周围的元素比较确定,因为它每一步都在寻找目标值,如果没找到,最后就返回 left 所在的位置。
代码及注解
class Solution {
public int searchInsert(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) {
right = mid - 1;
} else {
left = mid + 1;
}
}
return left;
}
}