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

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

leetcode 34

题目描述

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

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

输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]
nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]
输入：nums = [], target = 0
输出：[-1,-1]

题目分析

排序数组+搜索元素 ==>二分查找
先二分查找元素的第一个位置，然后若使用指针指向该节点进行遍历得到最后一个元素的话，，时间会比较长，因为不确定数组会很长，所以也可以通过二分查找找最后一个位置。

对于找目标的第一个位置，即左边界：
如果target==nums[mid]，即中间元素等于目标值，但是不能确定该中间元素是否为第一个位置，所以还需要继续往前查找 right = mid - 1;
如果target < nums[mid]，往前找，right = mid - 1;
所以小于等于情况可以合并

对于找最后一个位置（右边界），也用二分查找法，与左边界的查找刚好相反。
如果target==nums[mid]，即中间元素等于目标值，但是不能确定该中间元素是否为最后一个位置，所以还需要继续往有查找 left = mid + 1;
如果target > nums[mid]，往右找，left = mid + 1;
所以大于等于情况可以合并

class Solution {
    public int[] searchRange(int[] nums, int target) {
        //升序数组+查找元素==》二分查找
        //先二分查找元素的第一个位置，然后若使用指针指向该节点进行遍历得到最后一个元素的话，，时间会比较长，因为不确定数组会很长，所以也可以通过二分查找找最后一个位置
        int lower = lowerBound(nums,target);
        int upper = upperBound(nums,target);
        if(lower > upper) return new int[]{-1,-1};
        return new int[]{lower, upper};
    }

        //查找左边界
        public int lowerBound(int[] nums,int target){
            int left = 0, right = nums.length - 1;
            int mid = 0;
            while(left <= right){
                mid = left + (right - left) / 2;
                //if(target == nums[mid]){
                    //不能确定这个target是否为第一个位置，所以不能直接返回nums[mid],需要继续往前找right=mid-1;
                //}
                //所以，小于等于可以合并在一起
                if(target <= nums[mid]){
                    right = mid - 1;
                }
                else if(target > nums[mid]){
                    left = mid + 1;
                }
            }
            return left;
        }

        //查找右边界
        //跟查找下边界刚好相反,即如果等于也不能确定是最后一个，需要继续想右找，就可以将大于等于合并在一起
        public int upperBound(int[] nums,int target){
            int left = 0, right = nums.length - 1;
            int mid = 0;
            while(left <= right){
                mid = left + (right - left) / 2;
                if(target >= nums[mid]){
                    left = mid + 1;
                }
                else if(target < nums[mid]){
                    right = mid - 1;
                }
            }
            return right;
        }
    
}