LeetCode第34题-在排序数组中查找元素-java实现-图解思路与手撕代码
一、题目描述
给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。
二、解题思路与代码实现
1.解题思路
最自然的想法是先二分查找找到目标数字,然后向两边延展确定左右边界,但是题目要求用二分查找的方法解决该问题。
由题可知,寻找连续的目标数字范围即是分别找到第一次出现目标数字和第一次出现大于目标数字的位置。
为了代码复用,编写函数binarySearch,lower为true表示找第一次出现目标数字位置,lower为false表示找第一次出现大于目标数字位置。
在函数binarySearch中,由于要寻找第一次出现xx的位置(xx指目标数字或是大于目标数字),所以,右指针移动的时候更新ans,移动左指针的时候不更新。
需要注意的是,有可能出现[1],1的情况,ans的初始值可以设为数组长度。
2.代码实现
代码如下(示例):
private static int[] searchRange(int[] nums, int target) {
int[] res = new int[]{-1,-1};
if(nums.length==0){
return res;
}
int left=binarySearch(nums,target,true);
int right=binarySearch(nums,target,false)-1;
if (right>=left&&nums[right]==target&&nums[left]==target){
return new int[]{left,right};
}
return res;
}
private static int binarySearch(int[] nums, int target, boolean lower) {
int l=0,r=nums.length-1,mid=0,ans=nums.length;
while (r>=l){
mid=(r+l)/2;
if (target<nums[mid]||(lower&&nums[mid]==target)){
r=mid-1;
ans=mid;
}else {
l=mid+1;
}
}
return ans;
}
总结
这道题的解法对题目的理解和代码复用都非常巧妙,寻找连续的目标数字就是找到第一次出现该数字的位置和第一次出现大于该数字的位置,然后通过一个判断符号lower将两种情况联系在一起。