给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
进阶:
你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗?
示例 1:
输入:nums = [5,7,7,8,8,10], target = 8
输出:[3,4]
示例 2:
输入:nums = [5,7,7,8,8,10], target = 6
输出:[-1,-1]
示例 3:
输入:nums = [], target = 0
输出:[-1,-1]
二分法查找,不能再简单了。我居然又卡壳了。这个思路简单的不能再简单了,使用模板库就真的没有意思了,所以我自己手撸的,然后也写了一份模板库的,模板库的是真的没有意思,u1s1。
- 二分法查找位置
- 分别左右移动寻找端值
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> pos;
pos.push_back(-1);
pos.push_back(-1);
if(nums.size() == 0)
return pos;
//二分法
int low = 0;
int hight = nums.size() - 1;
int center = (low + hight) / 2;
while (low <= hight) {
center = (low + hight) / 2;
if (target > nums[center])
low = center + 1;
else if (target < nums[center])
hight = center - 1;
else break;
}
if(low>hight)
return pos;
else
pos[0] = pos[1] = center;
while(pos[0] - 1 >= 0 && nums[pos[0] - 1] == target)
pos[0]--;
while(pos[1] + 1 < nums.size() && nums[pos[1] + 1] == target)
pos[1]++;
return pos;
}
};
今天突然意识到,不能再那么浪费时间了,一是身体不好,二是时间真的不多了,很快就要工作了,现在连levelDB的源码都没看,已经三个星期了,自己都在折腾什么。
调用 STL 写法:
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> pos;
pos.push_back(-1);
pos.push_back(-1);
if(nums.size()==0)
return pos;
vector<int>::iterator it = lower_bound(nums.begin(), nums.end(), target);
if(it != nums.end() && *it == target ) {
pos[0] = it - nums.begin();
pos[1] = upper_bound(nums.begin(), nums.end(), target) - nums.begin() -1;
}
return pos;
}
};
STL 使用的坑还是很多的,比如 low_bound 的返回值可能是 end, 所以 *it 会出现非法访问,这就需要限制一下。这个思路还是需要转变一下。