linqibin/leetcode

33. 搜索旋转排序数组

Opened this issue · 0 comments

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。

搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。

你可以假设数组中不存在重复的元素。

你的算法时间复杂度必须是 O(log n) 级别。

示例 1:

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

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

思路:题目要求复杂度为O(log n),所以不能暴力遍历。采用二分法,将数组分成两半。每次分开后的子数组必有一半为升序数组,另一半是升序数组或乱序数组。若数组是升序,则可以靠边界元素来判断目标是否在数组内。每次可以丢弃一半数组不检查,则不需要完全遍历数组。

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number}
 */
var search = function(nums, target) {
    
    const stack = [];
    let start = 0, end = nums.length - 1;

    do{
        if (stack.length != 0) {
            let data = stack.pop();
            start = data.start;
            end = data.end;
        }

        let index = Math.floor((end + start) / 2);
        if(nums[index] == target) return index;
        
        
        if( nums[end] > nums[start] && nums[end] >= target && nums[start] <= target ){//如果数组是升序 且目标在里面
            if(index -1 >= start && nums[index -1] >= target){
                stack.push({start : start, end : index - 1})
            }
            if(nums[index + 1] <= target){
                stack.push({start : index + 1, end : end});
            }

        }else if(nums[end] < nums[start]){//或数组是乱序的
            if(index -1 >= start) stack.push({start : start, end : index - 1});          
            stack.push({start : index + 1, end : end});
        }
    }while(stack.length);

    return -1;
};