LeetCode第33题:搜索旋转排序数组

294 阅读1分钟

这是我参与更文挑战的第11天,活动详情查看: 更文挑战

题干

整数数组 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

解法:二分

这道题将数组旋转以后得到了两个升序的数组,所以我找到了旋转处,将数组一分为二,然后target的值如果大于旋转出的值就在arr2中二分寻找,如果target小于旋转出的值就在arr1中二分寻找

代码实现:

执行用时:76 ms, 在所有 JavaScript 提交中击败了94.75%的用户

内存消耗:38.7 MB, 在所有 JavaScript 提交中击败了95.00%的用户

  var search = function (nums, target) {
    // 将数组在旋转处分成两个升序数组,在根据target的值来判断target在哪一个数组中,之后进行二分查找
    let length = nums.length
    let theGapNumber
    for (let i = 0; i < length; i++) {
      if (nums[i] > nums[i + 1]) {
        theGapNumber = i + 1
      }
    }
    let arr1 = nums.slice(0, theGapNumber)
    let arr2 = nums.slice(theGapNumber, length)
    if (target > arr1[0]) {
      //使用arr1寻找
      return searchOneArea(arr1, target) == -1 ? -1 : searchOneArea(arr1, target)
    } else if (target < arr1[0]) {
      //使用arr2寻找
      let targetNum = searchOneArea(arr2, target)
      return targetNum == -1 ? -1 : (arr1.length + targetNum)
    } else {
      return 0
    }
  };
  var searchOneArea = function (nums, target) {
    let max = nums.length - 1;
    let min = 0;
    while (min <= max) {
      let mid = Math.floor((max + min) / 2)
      if (target == nums[mid]) {
        return mid
      } else if (target > nums[mid]) {
        min = mid + 1
      } else {
        max = mid - 1
      }
    }
    return -1
  }