题目描述:给定一个按照升序排列的整数数组 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]
算法思路:题目要求时间复杂度为O(logn),而且数组基本有序,所以首先想到的是二分法。二分法虽然思路不难,但是处理边界值的方面是极其棘手的,建议参看这篇文章连接
首先利用二分法找到第一个等于target数组的位置,然后再用一次二分法找到最后一个等于target的数组的位置。最后按照要求返回结果即可。
int left_boundary(int* nums, int numsSize, int target){
int low=0,high=numsSize-1;
while(low<=high){
int mid=(low+high)/2;
if(nums[mid]>target){
high=mid-1;//当nums[mid]大于target时,说明应该往数组小的一边找
}
else if (nums[mid]<target){
low=mid+1;//当nums[mid]小于target时,说明应该往数组大的一边找
}
else{
high=mid-1;当nums[mid]等于target时,而现在我们要找的是第一个等于target值得位置,所以继续网数组小的一边找
}
}
if(low<0||low>=numsSize||nums[low]!=target) return -1;
if(nums[low]==target) return low;
}
int right_boundary(int* nums, int numsSize, int target){
int low=0,high=numsSize-1;
while(low<=high){
int mid=(low+high)/2;
if(nums[mid]>target){
high=mid-1;
}
else if (nums[mid]<target){
low=mid+1;
}
else{
low=mid+1;
}
}
if(high<0||high>=numsSize||nums[high]!=target) return -1;
if(nums[high]==target) return high;
}
int* searchRange(int* nums, int numsSize, int target, int* returnSize){
*returnSize=2;
int left=left_boundary(nums,numsSize,target);
int *ans=(int*)malloc(sizeof(int)*2);
if(left==-1){
ans[0]=-1;
ans[1]=-1;
}else{
ans[0]=left;
ans[1]=right_boundary(nums,numsSize,target);
}
return ans;
}