终于知道两个模板的区别了!【手绘漫画】图解LeetCode之在排序数组中查找元素的第一个和最后一个位置(LeetCode 34)

2020-04-21 13:35:47 浏览数 (1)

首先看一下题目,

正常的二分查找,多了重复元素的限制,另外要求重复元素的头尾位置。

于是在不经意间,我发现了这个秘密。

3、正文

一起来看一下是什么秘密?

首先序列是 5、7、7、8、10,target=7,按照模板一二分查找。

代码语言:javascript复制
int left=0;
int right=nums.size()-1;
while(left<right){
    int mid=left right>>1;
    if(nums[mid]>=target){
        right=mid;
    }
    else{
        left=mid 1;
    }
}
return left;

可以看到,返回的是左侧的这个值,mid 是向下去整,right 也是,所以~

而模板二呢?

代码语言:javascript复制
int left=0;
int right=nums.size()-1;
while(left<right){
    int mid=(left right 1)>>1;
    if(nums[mid]<=target){
        left=mid;
    }
    else{
        right=mid-1;
    }
}
return right;

可以看到,返回的是右侧的这个值,mid 是向上去整,left 也是,所以~

最后返回的值写 left 或者 right 都是没问题的!因为 while 的结束条件是 leftright 相等。

4、代码

代码语言:javascript复制
class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        if(nums.empty()) return {-1,-1};

        int left=0;
        int right=nums.size()-1;
        while(left<right){
            int mid=(left right)>>1;
            if(nums[mid]>=target){
                right=mid;
            }
            else{
                left=mid 1;
            }
        }
        if(nums[left]!=target) return {-1,-1};
        int start=left;

        left=0;
        right=nums.size()-1;
        while(left<right){
            int mid=(left right 1)>>1;
            if(nums[mid]<=target){
                left=mid;
            }
            else{
                right=mid-1;
            }
        }
        if(nums[left]!=target) return {-1,-1};
        int end=right;
        
        return {start,end};
    }
};

0 人点赞