leetcode 34. 在排序数组中查找元素的第一个和最后一个位置

最新推荐文章于 2023-02-20 14:50:12 发布

Invisible Name

最新推荐文章于 2023-02-20 14:50:12 发布

阅读量53

点赞数

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/WIMPs_L/article/details/120442444

版权

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

进阶：

你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗？

示例 1：

输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]
示例 2：

输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]
示例 3：

输入：nums = [], target = 0
输出：[-1,-1]

提示：

0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109

class Solution {
public:
    int binarySearchLeft(int left,int right,vector<int>& nums, int target){
        if(left<0||right>=nums.size()||left>right){ //遗漏left>right，导致只有一个数{1}的时候，后面mid加一或减一溢出
            return -1;
        }
        int mid=(left+right)/2;
        if(target<nums[mid]){
            return binarySearchLeft(left,mid-1,nums,target);
        }
        else if(target>nums[mid]){
            return binarySearchLeft(mid+1,right,nums,target); //复制的上一句，导致写了-1，逻辑错误
        }
        else{
            if(mid==0){  //这部分之前和下面的else if写的或的逻辑，但mid-1溢出，就分开写，解决
                return mid;
            }
            else if(nums[mid-1]<target){
                return mid;
            }
            else{
                return binarySearchLeft(left,mid-1,nums,target);
            }
        }
    }

    int binarySearchRight(int left,int right,vector<int>& nums, int target){
        if(left<0||right>=nums.size()||left>right){  //遗漏left>right，导致只有一个数{1}的时候，后面mid加一或减一溢出
            return -1;
        }
        int mid=(left+right)/2;
        if(target<nums[mid]){
            return binarySearchRight(left,mid-1,nums,target);
        }
        else if(target>nums[mid]){
            return binarySearchRight(mid+1,right,nums,target); 
        }
        else{
            if(mid==nums.size()-1){
                return mid;
            }
            else if(nums[mid+1]>target){
                return mid;
            }
            else{
                return binarySearchRight(mid+1,right,nums,target);
            }
        }
    }

    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> result={-1,-1};
        int left=binarySearchLeft(0,nums.size()-1,nums,target);
        if(left==-1){
            return result;
        }
        int right=binarySearchRight(0,nums.size()-1,nums,target);
        result[0]=left;
        result[1]=right;
        return result;
    }
};