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

题目描述

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

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

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

思路

思路非常简单，就是单纯的二分查找，如果在数组中找到一个数和目标数相同，就在此位置向前和向后寻找，找到左边界和右边界返回即可。

代码

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

还有一种思路，未完待续…