All prompts are owned by LeetCode. To view the prompt, click the title link above.
First completed : December 17, 2024
Last updated : December 17, 2024
Related Topics : Array, Binary Search
Acceptance Rate : 38.4 %
I adapted this from my Q33 code to keep it at around average $O(log n) but due to the worst case scenarios e.g.
[1,1,1,1,1,1,1,1,1,1,1,1,1,1,...,1,1,2,1,1,1,1,1,...1,1,1]
this could be$O(n)$ . Should average out for random inputs though.
# adapted from my search in sorted array i answer
class Solution:
def search(self, nums: List[int], target: int) -> bool:
n = len(nums)
# find rotation amount via bin search
left, right = 0, n - 1
while left < right - 1 :
# if same vals
while left < right - 1 and left < n - 1 and nums[left] == nums[left + 1] :
left += 1
while left < right - 1 and right > 1 and nums[right] == nums[right - 1] :
right -= 1
# get mid index and val
mid = (left + right) // 2
val = nums[mid]
# ret if found
if target == val :
return True
# 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 True
# shift
if target < val :
right = mid - 1
continue
left = mid + 1
return False