推荐一篇不错的文章: 二分查找之美
二分查找
二分查找的代码是这样
int binarySearch(int* A, int n, int target){
int l = 0;
int r = n-1;
while(l<=r){
int mid = (l+r)>>1;
if(A[mid]==target) return mid;
if(A[mid]<target) l=mid+1;
else r=mid-1;
}
return -1;
}
循环不变式:
如果key存在于原始数组[0,n-1],那么它一定在[left,right]中。
初始化:
第一轮循环开始之前,处理的数组就是原始数组,这时显然成立。
保持:
每次循环开始前,key存在于待处理数组array[left, ..., right]中。
对于array[mid]<key,array[left, ..., mid]均小于key,key只可能存在于array[mid+1, ..., right]中;
对于array[mid]>key,array[mid, ..., right]均大于key,key只可能存在于array[left, ..., mid-1]中;
对于array[mid]==key,查找到了key对应的下标,直接返回。
在前两种情况中,数组长度每次至少减少1(实际减少的长度分别是mid-left+1和right-mid+1),直到由1(left==right)变为0(left>right),不会发生死循环。
终止:
结束时,left>right,待处理数组为空,表示key不存在于所有步骤的待处理数组,再结合每一步排除的部分数组中也不可能有key,因此key不存在于原数组。
二分查找左边界和右边界
其实就是leetcode的 Search for a range
我是这么写的
class Solution {
public:
vector<int> searchRange(int A[], int n, int target) {
//不如叫:二分查找左右断点
vector<int> range = {-1,-1};
int l=0;
int r=n-1;
int mid;
while(l<=r){
mid=(l+r)/2;
if(target<=A[mid]) r=mid-1;
else l=mid+1;
}
if(A[l]==target) range[0] = l;
l=0;
r=n-1;
while(l<=r){
mid=(l+r)/2;
if(target<A[mid]) r=mid-1;
else l=mid+1;
}
if(A[r]==target) range[1] = r;
return range;
}
};
问题是,为什么这样查找左边界的时候,跳出循环时, l就是要找的第一个元素的下标?
循环不变量:
如果 存在, [l,r+1]包含
最后退出循环的时候是不是有 l=r+1, 会不会是 l=r+2 ?