边工作边刷题:70天一遍leetcode: day 35-2

Search in Rotated Sorted Array

基本思路:根据mid的位置可以把rotated array分成两部分:顺序的和保持rotated,然后判断target的位置。target落入顺序部分的边界条件很简单,所以作为if部分。死记:mid比较在外,target比较在里。
错误点:

  • 虽然Search in Rotated Sorted Array是没有duplicate的,但是边界条件仍然有可能相等:
    • 判断是否是顺序的时候,如果只有2个元素,mid和low会是同一个元素,这时可能仍要继续判断:比如[3,1],1
    • 判断target的时候,target可能和low或者high的元素相等。
class Solution(object):
    def search(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        low,high=0,len(nums)-1
        while low<=high:
            mid = low + (high-low)/2
            if nums[mid]==target:
                return mid
            if nums[mid]>=nums[low]:
                if target<nums[mid] and target>=nums[low]:
                    high=mid-1
                else:
                    low=mid+1
            else:
                if target>nums[mid] and target<=nums[high]:
                    low=mid+1
                else:
                    high=mid-1
        
        return -1
                
原文地址:https://www.cnblogs.com/absolute/p/5678226.html