LeetCode 34. 搜索范围（search for a range）

题目描述

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

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

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

示例 1:

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

示例 2:

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

解题思路

利用二分查找的思想，分别找到数组中target出现的首位置和末位置，其中找首位置的步骤如下：

• 若数组中点值小于target，继续在中点值之后的数组查找
• 若数组中点值大于或等于target，说明前面数组中还可能有target，继续在中点值之前的数组查找
• 最后当首位置大于末位置时，首位置即为数组中第一个大于或等于target的值
• 若首位置上的数与target相等，则返回该位置，否则返回-1

同理可得到找末位置的步骤。

代码

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        if(nums.empty())
            return {-1,-1};
        int first=findFirst(nums,target);
        int last=findLast(nums,target);
        return {first,last};
    }
    int findFirst(vector<int>& nums, int target){
        int f=0,l=nums.size()-1;
        while(f<=l){
            int m=(f+l)/2;
            if(nums[m]<target)
                f=m+1;
            else
                l=m-1;
        }
        if(f<nums.size()&&nums[f]==target)
            return f;
        return -1;
    }
    int findLast(vector<int>& nums, int target){
        int f=0,l=nums.size()-1;
        while(f<=l){
            int m=(f+l)/2;
            if(nums[m]<=target)
                f=m+1;
            else
                l=m-1;
        }
        if(l>=0&&nums[l]==target)
            return l;
        return -1;
    }
};