给定一个按照升序排列的整数数组 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]
方法一:API
var searchRange = function(nums, target) {
return [nums.indexOf(target),nums.lastIndexOf(target)];
};
方法二:
线性查找:从头遍历找到第一个位置,从尾遍历找到最后一个位置。
var searchRange = function(nums, target) {
let res = [-1,-1];
let i = 0;
for(;i<nums.length;i++){
if(nums[i] == target){
res[0] = i;
break;
}
}
if(res[0] == -1) return res;
for(let j = nums.length-1;j>=i;j--){
if(nums[j] == target){
res[1] = j;
break;
}
}
return res;
};
复杂度分析
时间复杂度: O(n)。
空间复杂度: O(1)。线性扫描方法使用了固定大小的数组和几个整数,所以它的空间大小为常数级别的。
方法 三:二分查找
因为数组已经排过序了,我们可以使用二分查找的方法去定位左右下标。
var searchRange = function(nums, target) {
let n = nums.length;
let left = 0,right = n-1;
let index = -1;
while(left<=right){
let mid = (left+right)>>1;
if(nums[mid] > target){
right = mid - 1;
}else if(nums[mid] < target){
left = mid + 1;
}else{
index = mid;
break;
}
}
if(index == -1) return [-1,-1];
for(let i = index;i>=0;i--){
if(nums[i] !== target){
left = i+1;
break;
}
}
for(let j = index;j<n;j++){
if(nums[j] !== target){
right = j-1;
break;
}
}
return [left,right];
};
复杂度分析
时间复杂度:O(log n) 。由于二分查找每次将搜索区间大约划分为两等分,所以至多有 ⌈logn⌉ 次迭代。最坏为O(nlogn)。
空间复杂度:O(1) 。所有工作都是原地进行的,所以总的内存空间是常数级别的。