-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathm33.py
41 lines (33 loc) · 1.02 KB
/
m33.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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