leetCode34:在有序数组中查号元素的第一个和最后一个位置

目录

一、题目描述

二、解题思路

三、代码实现

---

一、题目描述

给定一个按照升序排列的整数数组 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;
}