leetcode81.搜索旋转排序数组 II

https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii

假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,0,1,2,2,5,6] 可能变为 [2,5,6,0,0,1,2] )。
编写一个函数来判断给定的目标值是否存在于数组中。若存在返回 true,否则返回 false。
示例 1:
输入: nums = [2,5,6,0,0,1,2], target = 0
输出: true
示例 2:
输入: nums = [2,5,6,0,0,1,2], target = 3
输出: false

这一题是leetcode33的升级版本,但是可以通过以下一段代码简化:

    if (left != right and nums[left]==nums[right]):
        left += 1
        while nums[left] == nums[left-1] and left < len(nums)-1:
            left += 1
        while nums[right] == nums[right-1] and right > 0:
            right -= 1

就是将左右相等的数先去掉,注意会有[1,1,1,1]全部是一个数字的测试用例,所以while循环要注意left和right不要越界。

有空list的测试用例,所以要加判断:

        if not nums:
            return False

最终通过版:

class Solution:
    def search(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: bool
        """
        if not nums:
            return False
        left, right = 0, len(nums)-1
        while left <= right:
            if (left != right and nums[left]==nums[right]):
                left += 1
                while nums[left] == nums[left-1] and left < len(nums)-1:
                    left += 1
                while nums[right] == nums[right-1] and right > 0:
                    right -= 1
            mid = (left+right) // 2
            if nums[mid] == target:
                return True
            if nums[mid] > nums[right]:  # 左有序
                if nums[mid] > target and nums[left] <= target:
                    right = mid - 1
                else:
                    left = mid + 1
            else:  # 右有序
                if nums[mid] < target and nums[right] >= target:
                    left = mid + 1
                else:
                    right = mid -1
        return False

你可能感兴趣的:(LeetCode,leetcode)