leetcode 33 搜索旋转排序数组

题目描述

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

( 例如，数组 [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

代码实现

//1. 先二分遍历找到分隔点index，特征是 < 前一个元素, >后一个元素;
//2. 把数组分成二个部分，[0,index-1], [index,length-1];
//3. 分别使用二分查找，找到给定的值。
//时间复杂度为log(n). 不确定有什么更好的办法

func search(nums []int, target int) int {
    if len(nums) == 0 {
        return -1
    }

    n := len(nums) - 1
    divisionIndex := findDivision(nums)
    if divisionIndex == 0 || divisionIndex == -1 {
        //非旋转排序数组
        return findTarget(nums, 0, n, target)
    }

    res := findTarget(nums, 0, divisionIndex-1, target)
    if res != -1 {
        return res
    }
    return findTarget(nums, divisionIndex, n, target)
}

//找到分割点
func findDivision(nums []int) int {
    low, high := 0, len(nums)-1
    for low < high {
        mid := low + (high-low)>>1

        if nums[high] < nums[mid] {
            low = mid + 1
        } else {
            high = mid
        }
    }
    return low
}

func findTarget(nums []int, low, high, target int) int {
    for low <= high {
        mid := low + (high-low)>>1
        if nums[mid] > target {
            high = mid - 1
        } else if nums[mid] < target {
            low = mid + 1
        } else {
            return mid
        }
    }
    return -1
}

思路

• 先二分遍历找到分隔点index，特征是 < 前一个元素, >后一个元素;
• 把数组分成二个部分，[0,index-1], [index,length-1];
• 分别使用二分查找，找到给定的值。

GitHub

• 源码传送门
• 各种数据结构及算法实现, LeetCode解题思路及答案
• 大厂面试题汇总及答案解析

参考资料

leetcode 33. 搜索旋转排序数组

本文为原创文章，转载注明出处，欢迎扫码关注公众号 楼兰 或者网站https://lovecoding.club,第一时间看后续精彩文章，觉得好的话，顺手分享到朋友圈吧，感谢支持。