33. 搜索旋转排序数组
题目描述
整数数组 nums 按升序排列,数组中的值 互不相同 。
在传递给函数之前,nums 在预先未知的某个下标 k(0 <= k < nums.length)上进行了 旋转,使数组变为 [nums[k], nums[k+1], …, nums[n-1], nums[0], nums[1], …, nums[k-1]](下标 从 0 开始 计数)。例如, [0,1,2,4,5,6,7] 在下标 3 处经旋转后可能变为 [4,5,6,7,0,1,2] 。
给你 旋转后 的数组 nums 和一个整数 target ,如果 nums 中存在这个目标值 target ,则返回它的下标,否则返回 -1 。
示例 1:
输入:nums = [4,5,6,7,0,1,2], target = 0
输出:4
示例 2:
输入:nums = [4,5,6,7,0,1,2], target = 3
输出:-1
示例 3:
输入:nums = [1], target = 0
输出:-1
提示:
- 1 <= nums.length <= 5000
- -10^4 <= nums[i] <= 10 ^4
- nums 中的每个值都 独一无二
- 题目数据保证 nums 在预先未知的某个下标上进行了旋转
- -10^4 <=target <=10 ^4
思路
二分查找法找出边界
原数组是升序排列的,数组的值互不相同,旋转排序数组的边界主要是:当前元素值大于后一个元素值,此元素的索引就是边界
let left = 0;
let n = nums.length;
let right = n - 1;
let border = 0;
while(left < right){
let mid = (left + right) >>> 1;
//边界值在右边 左指针右移
if(nums[mid] > nums[left]){
left = mid;
}
//边界值在左边 右指针左移
else{
right = mid;
}
if(nums[mid] > nums[mid + 1]){
//当前元素值大于后一个元素值,此元素的索引就是边界
border = mid;
break;
}
}
二分查找法找出目标值
根据 border 边界将数组一分为二,两部分皆是升序,做一次二分查找
-
target > nums[n - 1] 说明target在左边部分
左指针指向 0 右指针指向 border -
target < nums[n-1] 说明target在右边部分
左指针指向 border+1 右指针指向 n-1
完整代码
/**
* @param {number[]} nums
* @param {number} target
* @return {number}
*/
var search = function(nums, target) {
let left = 0;
let n = nums.length;
let right = n - 1;
let border = 0;
while(left < right){
let mid = (left + right) >>> 1;
if(nums[mid] > nums[left]){
left = mid;
}
else{
right = mid;
}
if(nums[mid] > nums[mid + 1]){
border = mid;
break;
}
}
if(target === nums[border]) return border;
if(target > nums[n - 1]){
left = 0;
right = border;
}else if(target < nums[n - 1]){
left = border + 1;
right = n - 1;
}else{
return n-1;
}
while(left <= right){
mid = (left + right) >>> 1;
if(nums[mid] > target){
right = mid - 1;
}else if(nums[mid] < target){
left = mid + 1;
}else{
return mid;
}
}
return -1;
};
向大佬学习
自己的思路实现结果为:
执行时间:56ms 超过99.735%的提交
无意间看执行时间更短的实现思路,作为菜鸡的我甚是惊叹
有位大佬通过,找target值的同时判断边界情况,直接修改数组元素的方式,直接二分查找暴力解题,效率极高
代码实现
/**
* @param {number[]} nums
* @param {number} target
* @return {number}
*/
var search = function(nums, target) {
let left = 0;
let right = nums.length-1;
while(left <= right) {
const mid = Math.floor(left + (right-left)/2);
if(nums[mid] == target) return mid;
if(target >= nums[0]) {
if(nums[mid] < nums[0]) {
nums[mid] = 10001;
}
} else {
if(nums[mid] >= nums[0]) {
nums[mid] = -10001;
}
}
if(nums[mid] < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
return -1;
};