给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。
示例:
输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
解决方案:如果不考虑 时间复杂度,那么可以用数组和两个count来记录下标,到时候输出就可以了,但是要求时间复杂度是log n 因此采用二分查找法
首先定义两个指针变量,分别存储左右两个位置的下标。首先去找目标值的最左面的下标,通过一个循环,每次取mid和目标值进行比较如果小于目标值的话,左侧的指针移动时+1。在循环结束的时候判断一下数组中是否包括目标值,不包括的话直接退出。右面的跟左侧相同,只不过正好相反。
那么相应的代码如下
#include <iostream>
using namespace std;
void Search_Target(int a[],int target)
{
int b[]={-1,-1};
int left=0;
int right=5;
int l=left;
int r=right;
int mid;
//第一次循环要找左边的那个目标值
while(left<right)
{
mid=(left+right)/2;//这样能保证找到的数在左边
if(a[mid]<target)
left=mid+1;
else
right=mid;
}
if(left>right||a[left]!=target)//说明走完一遍循环连目标值都没有找到
cout<<b[0]<<" "<<b[1];
return;
//不然的话,上边那一次循环已经找到了左边目标值的下标位置
//再通过一次循环来找右边的目标值的下标位置
while(l<r)
{
mid=(l+r)/2+1;//这样能保证找到的数在左边
if(a[mid]>target)
r=mid-1;
else
l=mid;
}
if(left>right||left>r)//说明走完一遍循环连目标值都没有找到或者此时找下的左边比右边大,那么久不存在
cout<<b[0]<<" "<<b[1];
else
{
b[0]=left;
b[1]=r;
cout<<b[0]<<" "<<b[1];
}
}
int main()
{
int a[]={5,7,8,8,8,10};
Search_Target(a,8);
cout<<endl;
return 0;
}
读者可自行画图理解;