排序数组中查找元素的第一个和最后一个位置
给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。
class Solution { public int[] searchRange(int[] nums, int target) { int n =nums.length; int le = binary(nums,target,true); int ri = binary(nums,target,false)-1; if(le<=ri&&ri<n){ return new int []{le,ri}; } else{ return new int[]{-1,-1}; } } public int binary(int[]nums,int target,boolean flag){ int l = 0; int r = nums.length-1; int ans =nums.length; while(l<=r){ int mid =(r+l)>>1; if((nums[mid]>=target&&flag)||(nums[mid]>target)){ ans = mid; r =mid-1; } else{ l = mid+1; } } return ans; } }
nums[mid]>target 找到第一个比target大的数字 找不到返回nums.length;
nums[mid]>=target&&flag找到第一个等于target的数字,如果不存在,返回的是第一个比target大的数字