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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/fuqiuai/article/details/83419584

题目链接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};
    }
};

猜你喜欢

转载自blog.csdn.net/fuqiuai/article/details/83419584