问题:
二分法涉及很多的边界条件,逻辑简单,就是写不好。
例如到底是
while(left < right)
还是while(left <= right)
,到底是right = middle
呢,还是要right = middle - 1
呢?
这里弄不清楚主要是因为对区间的定义没有想清楚,这就是不变量。
要在二分查找的过程中,保持不变量,这也就是循环不变量 (感兴趣的可以查一查)。
以下面的题为例子:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2
示例 2:输入: nums = [1,3,5,6], target = 2
输出: 1示例 3:
输入: nums = [1,3,5,6], target = 0
输出: 0
示例 4:输入: nums = [1], target = 0
输出: 0
二分法第一种写法:
①:以下的target是定义在一个左闭右闭的区间里,也就是[left,right]
while(left <= right) right = middle - 1
class Solution {
public int searchInsert(int[] nums, int target) {
int left=0;
int right=nums.length-1;//定义target在左闭右闭的区间[left,right]
while(left<=right){//当left==right,区间[left,right]依然有效
int mid=left+(right-left)/2;//防止溢出
if(nums[mid]>target){
right=mid-1;//target在左区间,所以[left,mid-1]
}else if(nums[mid]<target){
left=mid+1;//target在右区间,所以[mid+1,right]
}else{
return mid;
}
}
// 分别处理如下四种情况
// 目标值在数组所有元素之前 [0, -1]
// 目标值等于数组中某一个元素 return middle;
// 目标值插入数组中的位置 [left, right],return right + 1
// 目标值在数组所有元素之后的情况 [left, right], return right + 1
return right + 1;
}
}
二分法第二种写法:
①:定义 target 是在一个在左闭右开的区间里,也就是[left, right) 。
那么二分法的边界处理方式则截然不同。不变量是[left, right)的区间,如下代码可以看出是如何在循环中坚持不变量的。
while (left < right) right=mid
class Solution {
public int searchInsert(int[] nums, int target) {
int left=0;
int right=nums.length;//定义target在左闭右开的区间[left,right)
while(left<right){//当left==right时候,在[left,right)是无效的
int mid=left+(right-left)/2;//防止溢出
if(nums[mid]>target){
right=mid;//target在左区间,在[left,mid)中
}else if(nums[mid]<target){
left=mid+1;//target在右区间,在[mid+1,right)中
}else{
return mid;
}
}
// 分别处理如下四种情况
// 目标值在数组所有元素之前 [0,0)
// 目标值等于数组中某一个元素 return middle
// 目标值插入数组中的位置 [left, right) ,return right 即可
// 目标值在数组所有元素之后的情况 [left, right),return right 即可
return right;
}
}
总结:
二分法写不好,对区间定义不清楚
确定要查找的区间到底是左闭右开[left, right),还是左闭又闭[left, right],这就是不变量。
然后在二分查找的过程中,坚持循环不变量的原则。