在这篇博客中,我们以leetcode162题为例对二分查找中的易错点进行了辨析,但是在这篇博客中,关于左右边界的查找写的并不好。下面对左右边界的查找的代码进行简化。
左边界查找
在上篇博客中,这块关于左边界查找的通用模板是以[0,nums.length)为范围进行查找的。这篇博客中关于查找范围的讲解时非常棒的。下面对于其中模板的改进也是基于这个理论的:
int len=nums.length;
int mid=0;
int left=0,right=len-1;
while (left<=right){//①
mid=(left+right)/2;
if (nums[mid]>=target){
right=mid-1;//②
}else if (nums[mid]<target){
left=mid+1;
}
}
return left;
根据上篇博客的范围理论,首先,我们根据查找范围[0,nums.length-1],所以①处是小于等于。因为这个条件的存在,我们整个查找过程都是在闭区间中进行的。为了查找左边界,我们需要不断缩小右边界,反映在代码中就是②处。最终需要的返回值就是left。
右边界查找
与上面左边界查找类似,右边界的查找也是基于上篇博客中的范围理论。
int len=nums.length;
int mid=0;
int left=0,right=len-1;
while (left<=right){//①
mid=(left+right)/2;
if (nums[mid]>target){
right=mid-1;
}else if (nums[mid]<=target){
left=mid+1;//②
}
}
return right;
在查找右边界的过程中,要不断增大左边界。如②处所示。最终返回的right就是右边界。
对于leetcode中剑指offer的53-I题为例,其代码可以写为:
//找左边界的写法
public int search(int[] nums, int target) {
int cnt=0;
int len=nums.length;
int mid=0;
int left=0,right=len-1;
while (left<=right){
mid=(left+right)/2;
if (nums[mid]>=target){
right=mid-1;
}else if (nums[mid]<target){
left=mid+1;
}
}
while (left<len&&nums[left++]==target) {
cnt++;
}
return cnt;
}
//找右边界的写法
public int search(int[] nums, int target) {
int cnt=0;
int len=nums.length;
int mid=0;
int left=0,right=len-1;
while (left<=right){
mid=(left+right)/2;
if (nums[mid]>target){
right=mid-1;
}else if (nums[mid]<=target){
left=mid+1;
}
}
while (right>=0&&nums[right--]==target) {
cnt++;
}
return cnt;
}