Lintcode_62.搜索旋转排序数组

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Lintcode_62.搜索旋转排序数组相关的知识,希望对你有一定的参考价值。

假设有一个排序的按未知的旋转轴旋转的数组(比如,0 1 2 4 5 6 7 可能成为4 5 6 7 0 1 2)。给定一个目标值进行搜索,如果在数组中找到目标值返回数组中的索引位置,否则返回-1。

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

样例

给出[4, 5, 1, 2, 3]和target=1,返回 2

给出[4, 5, 1, 2, 3]和target=0,返回 -1

 

考虑O(logN)复杂度,用二分法,重点在于如何放缩区间,也就是start 和 end的赋值

当有序数组旋转后,某点 < A[0] 等价于在右边序列 , >A[0]等价于在左边序列

所有情况分析如下:

mid 在左序列 ,target > [mid] 则 target一定在mid右边。start = mid

mid 在左序列,target < [mid] 时有两种情况,若target 在左序列,则target 在mid左边,end = mid

                    若target在右序列,则target在mid 右边,start = mid

右序列同理分析,

mid 在右序列 ,target < [mid] 则 target一定在mid 左边。end = mid

mid 在右序列,target > [mid] 时有两种情况,若target 在左序列,则target 在mid左边,end = mid

                    若target在右序列,则target在mid 右边,start = mid

class Solution:
    """
    @param: A: an integer rotated sorted array
    @param: target: an integer to be searched
    @return: an integer
    """
    def search(self, A, target):
        # write your code here
        if not A:
            return -1
start, end = 0, len(A) - 1 if A[start] > A[end]: #有旋转 while start + 1 < end: mid = (start + end) / 2 if (A[mid] > A[-1] and target > A[mid]) or (A[mid] > A[-1] and target < A[mid] and target <= A[0]) or (A[mid] < A[0] and target > A[mid] and target <= A[0]): start = mid else: end = mid else:#无旋转,当做一般有序数组搜索 while start + 1 < end: mid = (start + end) / 2 if A[mid] < target: start = mid elif A[mid] > target: end = mid else : return mid if A[start] == target: return start elif A[end] == target: return end else: return -1

九章算法参考,更为简洁但是有点难理解:

class Solution:
    """
    @param A : a list of integers
    @param target : an integer to be searched
    @return : an integer
    """
    def search(self, A, target):
        # write your code here
        l, h = 0, len(A) - 1
        while (l <= h):
            m = l + ((h - l) >> 1)
            if A[m] == target:
                return m
            elif (A[m] > A[l] and target < A[m] and target >= A[l]) or (A[m] < A[l] and not (target <= A[h] and target > A[m])):
                h = m - 1
            else:
                l = m + 1
        return -1

 

以上是关于Lintcode_62.搜索旋转排序数组的主要内容,如果未能解决你的问题,请参考以下文章

62. 搜索旋转排序数组

lintcode_177_把排序数组转换为高度最小的二叉搜索树

63 搜索旋转排序数组II

lintcode-寻找旋转排序数组中的最小值 java 需复习

63. 搜索旋转排序数组 II

LintCode Python 简单级题目 39.恢复旋转排序数组