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

题目描述

给定一个按照升序排列的整数数组 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]

问题分析

此题还是用两次二分查找，第一次二分找出左端点，第二次二分找出右端点。找左端点时如果nums[mid]小于target，则说明待查找的左端点在不包括mid的右侧，否则在包括mid的左侧。然后判断找出的值是否等于target，如果相等则加入ans数组左端点的下标，如果不相等说明数组中不存在target，将-1加入ans数组。找右端点时如果nums[mid]大于target，则说明待查找的右端点在不包括mid的左侧，否则在包括mid的右侧。然后判断找出的值是否等于target，如果相等则加入ans数组右端点的下标，如果不相等说明数组中不存在target，将-1加入ans数组。最后返回ans数组。

代码实现

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> ans;
        int left;
        int right;
        int l = 0;
        int r = nums.size() - 1;
        int mid;
        if(nums.size() == 0){
            ans.push_back(-1);
            ans.push_back(-1);
            return ans;
        }   
        while(l < r){
            mid = l + r >> 1;
            if(nums[mid] < target)
                l = mid + 1;
            else
                r = mid; 
        }
        left = nums[l] == target? l : -1;
        ans.push_back(left);
        l = 0;
        r = nums.size() - 1;
        while(l < r){
            mid = l + r + 1 >> 1;
            if(nums[mid] > target)
                r = mid - 1;
            else
                l = mid; 
        }
        right = nums[l] == target? l : -1;
        ans.push_back(right);
        return ans;   
    }
};