Search for a Range 搜索范围

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

思路:

这道题两个关键点:数字是有序的,时间复杂度为O(logn)。

题目可以转化为两个子问题,1求排序数组中出现的第一个等于target的数字下标,2求排序数组中出现的最后一个等于target的数字下标。

具体两个字问题参考这两篇博文(第一个大于等于target的数字,最后一个等于target的数字之类都属于二分查找的变形版,应该牢记常见的变形版本和解答方法)

https://blog.csdn.net/ebowtang/article/details/50770315

循环不变式下的二分法,http://www.cnblogs.com/wuyuegb2312/archive/2013/05/26/3090369.html

求出下标以后,直接填写到res中即可,如果没有找到就填写-1

代码如下:

    vector<int> searchRange(vector<int>& nums, int target) {
	vector<int> res;
        if(nums.size()<=0){
            return {-1,-1};
        }
	int left = 0;
	int right = nums.size() - 1;
	while (left < right) {
		int mid = left + ((right - left) >> 1);
		if (nums[mid] < target) {
			left = mid + 1;
		}
		else {
			right = mid;
		}
	}
	if (nums[left] == target) {
		res.push_back(left);
	}
	else {
		res.push_back(-1);
	}


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



猜你喜欢

转载自blog.csdn.net/qq_26410101/article/details/80666920