「前端刷题」33. 搜索旋转排序数组

191 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。

题目

整数数组 nums 按升序排列,数组中的值 互不相同

在传递给函数之前,nums 在预先未知的某个下标 k0 <= 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

 

**进阶:**你可以设计一个时间复杂度为 O(log n) 的解决方案吗?

解题思路

二分法

刚刚接触这道题时,并不能理解他要做什么,既然要查找目标元素在数组中的索引,直接indexOf不就完事了吗,有啥好设计算法的,但是注意题目中的时间复杂度O(logn),indexOF的时间复杂度是O(n),明显不符合题意,所以很自然的能想到二分查找算法的运用。

此题还有一个注意点就是再执行二分时,肯定存在一部分是有序的。

const search = (nums, target) => {
    let [left, right] = [0, nums.length - 1];
    while (left <= right) {
        const mid = (left + right) >> 1;
        if (nums[mid] === target) return mid;
        if (nums[left] < nums[mid]) {
            // 左边是升序的
            if (nums[left] <= target && target <= nums[mid]) {
                // target在升序的里面
                right = mid - 1;
            } else {
                // target不在升序的里面
                left = mid + 1;
            }
        } else {
            // 右边升序
            if (nums[mid] <= target && target <= nums[right]) {
                // target在升序的里面
                left = mid + 1;
            } else {
                // target不在升序的里面
                right = mid - 1;
            }
        }
    }
    return nums[left + 1] === target ? left + 1 : -1;
};