33. Search in Rotated Sorted Array

[hayashi-ay]

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

[shining-ai]

smallest_index が分かっているなら、targetがpivotの左右どちらかにあるか判断できると思います。

nums[smallest_index] <= target <= nums[-1] 

なら右側にありませんか?

[hayashi-ay]

ありがとうございます。たしかにそうです。
書き直してみました。

def search(self, nums: List[int], target: int) -> int:
    def find_target(left: int, right: int, target: int) -> int:
        while left <= right:
            mid = (left + right) // 2
            if nums[mid] == target:
                return mid
            elif nums[mid] > target:
                right = mid - 1
            else:
                left = mid + 1
        return -1

    left, right = 0, len(nums) - 1
    while left < right:
        mid = (left + right) // 2
        if nums[mid] > nums[-1]:
            left = mid + 1
        else:
            right = mid
    smallest_index = left
    if nums[smallest_index] <= target <= nums[-1]:
        return find_target(smallest_index, len(nums) - 1, target)
    else:
        return find_target(0, smallest_index, target)