_33. Search in Rotated Sorted Array.md

33. Search in Rotated Sorted Array

All prompts are owned by LeetCode. To view the prompt, click the title link above.

Back to top

---

First completed : December 17, 2024

Last updated : December 17, 2024

---

Related Topics : Array, Binary Search

Acceptance Rate : 42.37 %

---

Solutions

• m33.py

Python

m33.py

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        n = len(nums)

        # find rotation amount via bin search
        left, right = 0, n - 1
        while left < right - 1 :
            # get mid index and val
            mid = (left + right) // 2
            val = nums[mid]

            # ret if found
            if target == val :
                return mid

            # shift
            if val < nums[left] :
                right = mid
                continue
            left = mid

        # account for no rotation
        offset = 0 if nums[right] - nums[left] > 0 else right

        # regular bin search
        left, right = offset, offset + n - 1
        while left <= right:
            mid = ((left + right) // 2)
            val = nums[mid % n]

            # found val
            if val == target :
                return mid % n

            # shift
            if target < val :
                right = mid - 1
                continue
            left = mid + 1

        return -1