题目:
给定一个按照升序排列的整数数组 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]
第一个出现的位置:
左边不是target右边是target。
最后一个出现的位置则是:
左边是target,右边不是。
这样我们就需要在二分查找的过程里,同时需要考虑 左边 和 右边,特征明显对应到了模板 3 .
我们可以用两次二分查找,分别找到第一个
和 最后一个
出现的位置。
class Solution {
public int[] searchRange(int[] nums, int target) {
int[] ans = new int[]{-1,-1};
if( nums.length==0 )return ans;
int i = 0, j = nums.length-1;
while( i+1<j ){
int mid = i+(j-i)/2;
if( nums[mid]>=target ){
j = mid;
}else{
i = mid;
}
}
if( nums[j]==target )ans[0] = j;
if( nums[i]==target )ans[0] = i;
i = 0;
j = nums.length-1;
while( i+1<j ){
int mid = i+(j-i)/2;
if( nums[mid]<=target ){
i = mid;
}else{
j = mid;
}
}
if( nums[i]==target )ans[1] = i;
if( nums[j]==target )ans[1] = j;
return ans;
}
}
注意跳出之后判断 i 和 j 的顺序。