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]

代码

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int x = -1, y = -1,len, low, high, mid, cut; 
        int[] res ={-1,-1};
        
        len = nums.length;
        if(len == 0) return res;
        low = 0;
        high = len-1;
        cut = 0;
            
        //找到其中一个target的位置
        while(low<=high){
            mid = (high-low)/2+low;
            if(nums[mid] == target){
                cut = mid;
                x = cut;
                y = cut;
                break;
            }
            else if(nums[mid] < target) low = mid+1;
            else high = mid-1;
        }
        
        if((cut==0 || cut==len-1) && nums[cut]!=target) return res;
        
        //以这个target为界，分成两个递增的数组，分别寻找上界和下界
        
        
        //1.寻找上界
        for(int i=cut-1; i>=0; i--){
            if(nums[i] == target) x = i;
            else break;
        }
        
        //2.寻找下界
        for(int i=cut+1; i<len; i++){
            if(nums[i] == target) y = i;
            else break;
        } 
        
        res[0] = x;
        res[1] = y;
        return res;
    }
}