前端算法(31)

55 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

题目

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

你必须设计一个时间复杂度为 O(log n) 的算法解决此问题。

输入: nums = [4,5,6,7,0,1,2], target = 0
输出: 4

题目解析

思路一

通过对比旋转数组中的起始值和中间值,可以确认有序递增区间在中间值的左边还是右边。注意考虑起始值与中间值相等的情况(数组只有两个值时。然后只需要对比target与递增区间的边界值来确定target所在的范围即可,注意考虑target与边界相等的情况。

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number}
 */
var search = function(nums, target) {
    let l = 0, r = nums.length-1, mid
    while(l <= r){
        mid = Math.floor((l+r)/2)
        if(nums[mid] == target) return mid
        if(nums[l] <= nums[mid]){ 
        // 左边有序递增
            if(nums[l] < target && target < nums[mid]){
                r = mid - 1
            } else if(nums[l] == target){ // 相等时
                return l
            } else { 
                l = mid + 1
            }
        } else { 
        // 右边有序递增
            if(nums[mid] < target && target < nums[r]){
                l = mid + 1
            } else if(nums[r] == target) { // 相等时
                return r
            } else {
                r = mid - 1
            }
        }
    }
    return -1
};

思路二

将数组分为两段有序数组,然后根据数组第一位的值和target比较,去查找其中包含了target的数组

var search = function (nums, target) {
    var findMin = (nums) => {
        var left = 0,
            right = nums.length - 1
        while (left < right) {
            var mid = left + (right - left >> 1)
            if (nums[mid] > nums[right]) {
                left = mid + 1
            } else {
                right = mid
            }
        }
        return left
    }
    var findTarget = (left, right, nums, target) => {
        while (left <= right) {
            var mid = left + (right - left >> 1)
            if (nums[mid] > target) {
                right = mid - 1
            } else if (nums[mid] == target) {
                return mid
            } else {
                left = mid + 1
            }
        }
        return -1
    }
    var minIndex = findMin(nums),
        first = nums[0]
    if (nums[0] < nums[nums.length - 1]) {
        return findTarget(0, nums.length - 1, nums, target)
    }
    if (first > target) {
        return findTarget(minIndex, nums.length - 1, nums, target)
    } else if (first == target) {
        return 0
    } else {
        return findTarget(1, minIndex, nums, target)
    }
};