day18-在排序数组中查找元素的第一个和最后一个位置
给定一个按照升序排列的整数数组 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
解题思路:
因为时间复杂度要达到O(log n)所以二分法来思考,因为整个数组是升序的,所以如果目标值在数组中是一个的话,那么问题就等同于二分法查找某个值的位置,如果目标值在数组中有多个,因为是升序,所以可以把多个目标值看成一个整体。判断左右边界是否都等于目标值,如果不满足条件,那么进行二分,判断中间数和目标值的大小从而缩小范围,如果中间值和目标值相等,那么边界值一定有一个和目标值不相等(都相等的话循环开始的if条件就满足了),所以我们只需要判断一边是否相等,相等的话就缩短另一边界值,直至两个边界值都是目标值为止。
解题代码:
class Solution {
public int[] searchRange(int[] nums, int target) {
int n = nums.length-1;
if(n==-1){
return new int[]{-1,-1};
}
if(n==0){
return nums[0]==target ? new int[]{0,0} : new int[]{-1,-1};
}
int l = 0;
int mid = 0;
while(l<=n){
if(nums[l]==target&&nums[n]==target){
return new int[]{l,n};
}
mid = (l+n)/2;
if(nums[mid]>target){
n = mid-1;
}else if(nums[mid]<target){
l = mid+1;
}
if(nums[mid]==target){
mid = nums[l]<target ? ++l:--n;//真正的目的不是为mid赋值,而是为了++l或者--n。
}
}
return new int[]{-1,-1};
}
}
执行结果: