Given a sorted array of integers, find the starting and ending position of a given target value.
Your algorithm's runtime complexity must be in the order of O(log n).
If the target is not found in the array, return [-1,
.
-1]
For example,
Given [5, 7, 7, 8, 8, 10]
and target value 8,
return [3, 4]
.
思路:这道题是binary search的改进版。当寻找到target所在的k位置时,需要判断target的起始索引值和终止索引值。可以这么思考:首先得到k的位置,如果k不存在则返回[-1,-1];如果k存在,则起始的索引值肯定是在k的左半侧,终止索引值在k的右半侧。然后对这两侧再进行二分查找。因此总共的时间复杂度为O(n)。
class Solution { public: vector<int> searchRange(int A[], int n, int target) { vector<int> res; res.clear(); if (n<=0) { res.push_back(-1); res.push_back(-1); return res; } int L = 0, R = n - 1, M; while(L <= R) { M = (L + R) / 2; if (A[M] > target) { R = M - 1; } else if(A[M] < target) { L = M + 1; } else { break; } } if (A[M] != target) { res.push_back(-1); res.push_back(-1); } else { L = 0, R = M - 1; int mid; while(R>=0 && A[R]==target) { mid = (L + R) / 2; if (A[mid] < target) { L = mid + 1; } else if(A[mid] >= target) { R = mid - 1; } } res.push_back(R + 1); L = M + 1,R = n - 1; while(L<n && A[L] == target) { mid = (L + R) / 2; if (A[mid] <= target) { L = mid + 1; } else { R = mid - 1; } } res.push_back(L-1); } return res; } };