二分法之通用模板

在这里我们一共有两个模板

模板1

我们将区间[l,r]划分为[l,mid]和[mid+1,r]时,其更新操作是r=mid或者l=mid+1;计算mid时不用加一。

int searchRange(vector<int>& nums, int target) {
        int l=0,r=nums.size()-1;
        while(l<r){
            int mid=l+r>>1;
            if(check(mid))r=mid;
            else l=mid+1;
        }
    return l;
}

模板2

我们将区间[l,r]划分成[l,mid-1]和[mid,r]时,更新操作是r=mid-1或者l=mid;这时为了防止死循环,计算mid时需要+1。因为普通的整型计算是向下取整的+1,相当于四舍五入。

int searchRange(vector<int>& nums, int target) {
        int l=0,r=nums.size()-1;
        while(l<r){
            int mid=l+r+1>>1;
            if(check(mid))l=mid;
            else r=mid-1;
        }
    return l;
}

这两个模板都可以找到想要我们找到的target。举个例子如下图我们想要找到9

 

 首先使用模板一

int searchRange(vector<int>& nums, int target) {
        int l=0,r=nums.size()-1;
        while(l<r){
            int mid=l+r>>1;
            if(nums[mid]>=target)r=mid;
            else l=mid+1;
        }
    return l;
}

这样我们可以看到,模板一找到的最左边的等于target的数。

因为我们总是完全抛弃左边,右边抛弃除nums[mid]的数所以我们选择check函数为nums[mid]>=target。当数组中有相同的元素的时候,如图所示,当数组中没有相同的元素的时候,当nums[r]==target时,nums[mid]<target,l会一步步向右缩直到l=r。

然后我们使用模板二

int searchRange(vector<int>& nums, int target) {
        int l=0,r=nums.size()-1;
        while(l<r){
            int mid=l+r+1>>1;
            if(nums[mid]<=target)l=mid;
            else r=mid-1;
        }
    return l;
}

这样我们可以看到,模板一找到的最右边的等于target的数。

因为我们总是完全抛弃右边,左边抛弃除nums[mid]的数所以我们选择check函数为nums[mid]>=target。当数组中有相同的元素的时候,如图所示,当数组中没有相同的元素的时候,当nums[l]==target时,nums[mid]>target,r会一步步向右缩直到l=r。

所以对于没有相同元素的时候,使用两种模板都可以找到target。对于存在相同元素的时候,按照题意,进行判断是要找第一个为target还是最后一个为target的使用两种不同的模板。

这里有一道例题来帮助大家消化。

 
posted @ 2019-07-14 18:02  fromzore  阅读(2161)  评论(0编辑  收藏  举报