题目
给定一个按照升序排列的整数数组 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(log n) ,用二分查找,注意边界条件
实现
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> res;
if (nums.size() == 0) {
res.push_back(-1);
res.push_back(-1);
return res;
}
int start = 0;
int end = nums.size() - 1;
if (nums[start] > target || nums[end] < target) {
res.push_back(-1);
res.push_back(-1);
return res;
}
while (start <= end){
if (start == end && target != nums[start]) {
res.push_back(-1);
res.push_back(-1);
break;
}
if ((start + 1) == end && target != nums[start] && target != nums[end]) {
res.push_back(-1);
res.push_back(-1);
break;
}
if (target == nums[start]) {
int temp1 = start;
int temp2 = start;
while(temp1 >= 0 && nums[temp1] == target) {
temp1--;
}
res.push_back(++temp1);
while(temp2 < nums.size() && nums[temp2] == target) {
temp2++;
}
res.push_back(--temp2);
break;
}
if (target == nums[end]) {
int temp1 = end;
int temp2 = end;
while(temp1 >= 0 && nums[temp1] == target) {
temp1--;
}
res.push_back(++temp1);
while(temp2 < nums.size() && nums[temp2] == target) {
temp2++;
}
res.push_back(--temp2);
break;
}
if (target > nums[start] && target < nums[end]) {
int mid = (start + end) / 2;
if (target >= nums[mid]) {
start = mid;
} else {
end = mid;
}
}
}
return res;
}
};