LeetCode第 34 题:在排序数组中查找元素的第一个和最后一个位置(C++)

34. 在排序数组中查找元素的第一个和最后一个位置 - 力扣(LeetCode)

就是二分查找的变体,感觉跟简单题差不多:

数据结构与算法之美:16 | 二分查找(下)

这儿怎么找都行,那就找第一个等于target的位置吧:

class Solution {
public:
    int bsearch(vector<int> &nums, int low, int high, int target){
        while(low <= high){
            auto mid = (low+high)>>1;
            if(nums[mid] > target)  high = mid-1;
            else if(nums[mid] < target) low = mid+1;
            else{
                if(mid == 0 || nums[mid-1] != target)   return mid;
                high = mid-1;
            }
        }
        return -1;
    }
    vector<int> searchRange(vector<int>& nums, int target) {
        int index = bsearch(nums, 0, nums.size()-1, target);
        if(index == -1) return {-1,-1};
        int i = index+1;
        while(i < nums.size() && nums[i] == nums[index])    ++i;
        return {index, i-1};
    }
};

也可以两次二分查找,第二次找最后一个就行。

猜你喜欢

转载自blog.csdn.net/qq_32523711/article/details/107789625