leetcode-24-在排序数组中查找元素的第一个和最后一个位置-medium

题目描述

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

• 一看当然是懵逼的，知道是用二分，但是不知道怎么用二分来寻找左右边界
• 看了题解发现二分的细节太多了，这篇题解讲的非常透彻点这里
• 思路很简单，细节是魔鬼
  题解代码：

vector<int> searchRange(vector<int>& nums, int target) {
	vector<int> ans;
	ans.push_back(left_bound(nums, target));
	ans.push_back(right_bound(nums, target));
	return ans;
}
int left_bound(vector<int> &nums, int target) {
	if (nums.size() == 0) return -1;
	int i = 0, j = nums.size(), mid = -1; //搜索区间是左闭右开的
	while (i < j) {
		//左半区间[i, mid) 右半区间[mid+1, j)
		mid = i + (j - i) / 2; //中间位置
		if (nums[mid] == target) { //收缩上界
			j = mid;
		}
		else if (nums[mid] > target) { //左半区间 [i, mid)
			j = mid;
		}
		else if (nums[mid] < target) { //右半区间 [mid+1, j) 
			i = mid + 1;
		}
	}
	if (i == nums.size()) return -1;
	else return (nums[i] == target) ? i : -1;
}
int right_bound(vector<int> &nums, int target) {
	if (nums.size() == 0) return -1;
	int i = 0, j = nums.size(), mid = -1; //搜索区间是左闭右开的
	while (i < j) {
		mid = i + (j - i) / 2;
		//左半区间 [i, mid) 右半区间 [mid+1, j)
		if (nums[mid] == target) { //收缩下界
			i = mid + 1;
		}
		else if (nums[mid] > target) { //左半区间
			j = mid;
		}
		else if (nums[mid] < target) { //右半区间
			i = mid + 1;
		}
	}
	if (j == 0) return -1;
	return nums[j - 1] == target ? (j - 1) : -1;
}

• 自写代码

vector<int> searchRange(vector<int>& nums, int target) {
	if(nums.size()==0) return {-1,-1};
	vector<int> ret;
	int l = 0;
	int r = nums.size();
	int mid;
	// [l,r)
	while(l<r){
		mid = (l+r)/2;
		if(nums[mid]==target){
			//[l,mid)
			r = mid;
		}else if(nums[mid]<target){
			//[mid+1,r]
			l = mid+1;
		}else{
			//[l,mid)
			r = mid;
		}
	} 
	if(l==nums.size()) ret.push_back(-1);
	else ret.push_back(nums[l]==target?l:-1);
	
	//------------------------------
	
	l = 0;
	r = nums.size();
	// [l,r)
	while(l<r){
		mid = (l+r)/2;
		if(nums[mid]==target){
			//[mid+1,r)
			l = mid+1;
		}else if(nums[mid]<target){
			//[mid+1,r]
			l = mid+1;
		}else{
			//[l,mid)
			r = mid;
		}
	} 
	
	if(r == 0) ret.push_back(-1);
	else ret.push_back(nums[r-1]==target?r-1:-1);
	
	
	return ret;
}