刷题学习记录——leetcode 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]
来源：力扣（LeetCode）

初步思路

遍历数组，如果和target相等就记录第一个位置的下标作为左标，最后一个和target相等的元素下标作为右标，然后返回即可，应该是o(n)复杂度。

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int left = -1,right = -1;
        int []ans = {-1,-1};
        for(int i = 0;i < nums.length;)
        {
            if(nums[i] < target)
                i++;
            else if(nums[i] == target)
            {
                if(ans[0] == -1)
                {
                   ans[0] = i;
                   ans[1] = i;
                }
                else
                {
                    ans[1] = i;
                }
                i++;
            }
            else if(nums[i] > target)
            {
                break;
            }
        }
        return ans;
    }
}

看了题解之后

为了达到o（logn）的复杂度，我们采用二分查找的方式查找第一个满足>=target的数（左标）和第一个大于target的数（减一为右标），返回即可。

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int left = binarySearch(nums,target,true);
        int right = binarySearch(nums,target,false) - 1;

		//判断条件
        if(left <= right && right < nums.length && nums[left] == target && nums[right] == target)
        {
            return new int[]{left,right};
        }
        return new int[]{-1,-1};
    }
	//二分查找，lower仅用于区分是查找左标还是右标
    public int binarySearch(int[] nums,int target, boolean lower){
            int left = 0,right = nums.length - 1,ans  = nums.length;
            while(left <= right){
                int mid = (left + right) / 2;
                if(nums[mid] > target || (lower && nums[mid] >= target)){
                    right = mid - 1;
                    ans = mid;
                }else{
                    left = mid + 1;
                }
            }
            return ans;
    }
}