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