给定一个包含 n 个整数的排序数组,找出给定目标值 target 的起始和结束位置。
如果目标值不在数组中,则返回[-1, -1]
样例
给出[5, 7, 7, 8, 8, 10]
和目标值target=8
,
返回[3, 4]
挑战
时间复杂度 O(log n)
class Solution {
/**
*@param A : an integer sorted array
*@param target : an integer to be inserted
*return : a list of length 2, [index1, index2]
*/
public:
vector<int> searchRange(vector<int> &A, int target) {
// write your code here
vector<int> result;
result.push_back(-1);
result.push_back(-1);
if(A.empty())
return result;
int n=A.size();
int begin=0,end=n-1,mid;
while(begin<end){
mid=(begin+end)/2;
if(A[mid]==target){
begin=mid;
break;
}
else if(A[mid]<target)
begin=mid+1;
else
end=mid-1;
}
int i,j;
if(A[begin]==target){
i=begin;
j=begin;
while(A[i]==target) i--;
while(A[j]==target) j++;
result[0]=i+1;
result[1]=j-1;
}
return result;
}
};
如有问题请留言。
转载请注明出处。