LeetCode34.在排序数组中查找元素的第一个和最后一个位置

1. 题目

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

2. 思路

由于题目要求时间复杂度为 O(log n) 级别，且数组 nums 为有序排列数组，因此考虑使用二分查找的方法来确定开始位置与结束位置。利用二分思想，先找其左边界，再找其右边界即可。

(1) 首先，找其左边界：

int left = 0;
int right = nums.size() - 1;
while(left < right)
{
	int mid = left + (right - left) / 2;
	if(nums[mid] >= target) {
		right = mid;
	}
	else {
		left = mid + 1;
	}
}
return left;

要寻找左边界，则当 nums[mid] > target 或者 nums[mid] = target 时，均应该向左半区寻找，且考虑到 mid 可能就是左边界的情况，因此 right = mid；当 nums[mid] < target 时，应该向右半区寻找，且 mid 一定不是左边界，因此 left = mid + 1。
因为 while 循环终止条件为 left = right ，此时锁定一个位置，则该位置即为要寻找的左边界位置，且此时返回 left 和 right 都可以。此时，mid向下取整，mid = left + (right - left) / 2。

(2) 然后，找其右边界：

int left = 0;
int right = nums.size() - 1;
while(left < right)
{
	int mid = left + (right + 1 - left) / 2;
	if(nums[mid] <= target) {
		left = mid;
	}
	else {
		right = mid - 1;
	}
}
return right;

要寻找右边界，则当 nums[mid] < target 或者 nums[mid] = target 时，均应该向右半区寻找，且考虑到 mid 可能就是右边界的情况，因此 left = mid；当 nums[mid] > target 时，应该向左半区寻找，且 mid 一定不是右边界，因此 right = mid - 1。
因为 while 循环终止条件为 left = right ，此时锁定一个位置，则该位置即为要寻找的右边界位置，且此时返回 left 和 right 都可以。此时，mid向上取整，mid = left + (right + 1 - left) / 2。

3. 题解

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        if(nums.empty()) return {-1,-1};
        
        //找左边界
        int left = 0;
        int right = nums.size()-1;
		while(left < right)
		{
			int mid = left + (right - left) / 2;
			if(nums[mid] >= target) {
				right = mid;
			}
			else {
				left = mid + 1;
			}
		}
		if(nums[left] != target) return {-1,-1};
		int start = left;

		//找右边界
		left = 0;
		right = nums.size()-1;
		while(left < right)
		{
			int mid = left + (right + 1 - left) / 2;
			if(nums[mid] <= target) {
				left = mid;
			}
			else {
				right = mid - 1; 
			}
		}
		if(nums[right] != target) return {-1,-1};
		int end = right;

		return {start,end};
    }
};

4. 结果