题目
给定一个按照升序排列的整数数组 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]
提示:
0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:二分查找
面对排序好的数组,应该想到二分查找
这里需要查找第一次出现和最后一次出现的数,然而因为数组是有序的,所以必定是连在一起的,所以找到其中一个向周围扩散查找
- 二分查找
- 两边扩散
二分查找的写法:
-
何时退出循环:
当rear指针<head指针时退出循环,也即rear指针必定在head指针之后。
当找到mid==target时退出循环
-
怎样的条件下维护head和rear指针:
当 mid指针的值>target时:维护rear,target在前面,尾指针提到前面
当mid指针的值<target时:维护head,target在后面,头指针提到后面
-
怎样维护指针:
要使找不到数值时,能够触发rear>head
所以,
rear=mid-1
head=mid+1
因为mid已经检查过不等于target,所以不需要指向mid
JAVA
public int[] searchRange(int[] nums, int target) {
{
if(nums.length==0) return new int[]{-1,-1};
if(nums[0]>target) return new int[]{-1,-1};
if(nums[nums.length-1]<target) return new int[]{-1,-1};
int head=0,rear=nums.length-1;
boolean t = true;
//找目标
while (rear>=head){
//取中值与目标比较
int mid=(head+rear)/2;
//目标大于中值,head=中值
if(target>nums[mid]){
head=mid+1;
}
//目标小于中值,rear=中值
else if(target<nums[mid]){
rear = mid-1;
}
//目标等于中值,退出循环,记录位置
else if(target==nums[mid]){
head=mid;
t=false;//找到
break;
}
}
//不存在目标
if(t) return new int[] {-1,-1};
//找到目标向两边寻找边界
rear=head;
do{
--head;
}while (head>=0&&nums[head]==target);
do{
++rear;
}while (rear<=nums.length-1&&nums[rear]==target);
return new int[] {head+1,rear-1};
}
}