目录
一、题目描述
给定一个按照升序排列的整数数组 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]
提示:
- 0 <= nums.length <= 105
- -109 <= nums[i] <= 109
- nums 是一个非递减数组
- -109 <= target <= 109
二、解题思路
这题是经典的二分查找题,可随机访问的数组,并且有序,二分的典型标志。
自然可以用经典的二分查找的代码实现,效率和简洁度都很好,唯一要注意的就是左右标记该如何变化,求不同的结果有不同的变化方式,也不难,注意一下即可。
由于最近刚看过STL库的upper_bound和lower_bound的源码实现,所以这里照着自己实现一遍,仿照的是源码的使用迭代器的版本,如果是下标的版本,跟经典的二分查找代码几乎没有区别。
三、代码实现
#include<bits/stdc++.h>
using namespace std;
//按照两个函数独立的功能分开来写了,本题可以写成一个函数,提高代码复用
int upperIndex(vector<int>& nums, int target) {
int n = nums.size();
int count = n, cur = 0;
while (count) {
//step代替经典的mid,count用来记录查找范围内的元素个数
//i是游标
int i = cur, step = count / 2;
i += step;
if (nums[i] <= target) {
//数组当前值不大于目标值,说明结果在右半边
//在右半边继续查找
cur = ++i;
count -= step + 1;
} else {
//在左半边
count = step;
}
}
return cur;
}
int lowerIndex(vector<int>& nums, int target) {
int n = nums.size();
int count = n, cur = 0;
while (count) {
int i = cur, step = count / 2;
i += step;
//写在一个函数里的话,这里的if可以合并
//传一个bool变量,为true就执行upper的功能,false执行lower的功能,如下代码
/*if (nums[i] < target || (flag && nums[i] <= target)) {
/*if (nums[i] < target || (flag && nums[i] <= target)) {
cur = ++i;
count -= step + 1;
}*/
if (nums[i] < target) {
cur = ++i;
count -= step + 1;
} else {
count = step;
}
}
return cur;
}
vector<int> searchRange(vector<int>& nums, int target) {
int n = nums.size();
if (n == 0) return { -1, -1 };
int left = lowerIndex(nums, target);
int right = upperIndex(nums, target);
if (left != n && nums[left] == target) return { left, right - 1 };
return { -1, -1 };
}
int main() {
vector<int> nums = { 2,2 };
int target = 3;
vector<int> res = searchRange(nums, target);
for (int i = 0; i < 2; ++i) {
cout << res[i] << " ";
}
// cout << upperIndex(nums, 4) << " ";
// cout << lowerIndex(nums, 4) << " ";
return 0;
}