LeetCode Search in Rotated Sorted Array II

LeetCode解题之Search in Rotated Sorted Array II

原题

把一个不降序的数组进行旋转，如[0,1,1,1,2,3,4,5]旋转3位成为[3,4,5,0,1,1,1,2]。在这样的数组中判断目标数字是否存在。

注意点：

• 不知道数组旋转了多少位

例子：

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

输出: True

解题思路

当时写 Search in Rotated Sorted Array 的时候已经考虑到数字重复的问题了，所以直接把原来的代码修改一下，返回下标改为返回布尔值。

AC源码

class Solution(object):
    def search(self, nums, target):
        """ :type nums: List[int] :type target: int :rtype: int """
        left = 0
        right = len(nums) - 1
        while left <= right:
            mid = left + (right - left) // 2
            if nums[mid] == target:
                return True
            if nums[mid] > target:
                if nums[left] <= target or nums[mid] < nums[left]:
                    right = mid - 1
                else:
                    left = mid + 1
            else:
                if nums[left] > target or nums[mid] >= nums[left]:
                    left = mid + 1
                else:
                    right = mid - 1
        return False


if __name__ == "__main__":
    assert Solution().search([4, 5, 5, 6, 7, 0, 1, 2], 4) == True
    assert Solution().search([4, 5, 6, 7, 7, 7, 7, 7, 0, 1, 2], 7) == True

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。