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

Bertha 。 2022-05-03 00:55 111阅读 0赞

**题目链接**：[https://leetcode-cn.com/problems/generate-parentheses/description/][https_leetcode-cn.com_problems_generate-parentheses_description]

#### 题目描述 ####

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

你的算法时间复杂度必须是 O(log n) 级别。

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

#### 示例 ####

输入: nums = \[5,7,7,8,8,10\], target = 8  
输出: \[3,4\]

输入: nums = \[5,7,7,8,8,10\], target = 6  
输出: \[-1,-1\]

#### 解决方法 ####

使用分治算法，进行**二分查找**

//使用分治算法，进行二分查找
    class Solution { 
    public:
        vector<int> searchRange(vector<int>& nums, int target) {    
            
            int l=0,r=nums.size()-1; //二分查找
            int pos=-1;
            while(l<=r){ 
                int m=(l+r)/2;
                if (nums[m]>target) r=m-1;
                else if (nums[m]<target) l=m+1;
                else {  pos=m; break;}
            }
            
            if (pos==-1) return { -1,-1}; //没找到
            
            int start=pos,end=pos; //找到了，则往前和往后分别找到起始位置
            while(start>=0 && nums[start]==target) start--;
            while(end<=nums.size()-1 && nums[end]==target) end++;
            return { start+1,end-1};
        }
    };

[https_leetcode-cn.com_problems_generate-parentheses_description]: https://leetcode-cn.com/problems/generate-parentheses/description/