力扣:34,在排序数组这种查找元素的第一个和最后一个位置

题目描述

给你一个按照非递减顺序排列的整数数组 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

来源:力扣(LeetCode)
题目链接:力扣34.在排序数组中查找元素的第一个和最后一个位置

扫描二维码关注公众号,回复: 14528792 查看本文章

解题思路

常规题,利用二分法解决即可,需要注意的是边界问题,比如数组2 3,l=0,r=1,mid=0,如果nums[mid]<=target的话,l=mid,这样就陷入了死循环,所以在这里我们求mid的值时让l+r再加上个1,避免出现死循环,当nums[mid]>=target时不会出现死循环问题。

代码实现

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int>v;
        v.push_back(-1);
        v.push_back(-1);
        int n=nums.size();
        int i=0,j=n-1;
        if(n==0)
        return v;
        while(i<j)
        {
            int mid=(i+j)>>1;
            if(nums[mid]>=target) j=mid;
            else
            i=mid+1;
        }
        if(nums[i]!=target) 
        return v;
        else
        {
            v.clear();
            v.push_back(i);
            i=0,j=n-1;
            while(i<j)
            {
                int mid=(i+j+1)>>1;
                if(nums[mid]<=target) i=mid;
                else
                j=mid-1;
            }
            v.push_back(i);
        }
        return v;

    }
};

猜你喜欢

转载自blog.csdn.net/qq_52905520/article/details/126850930