forked from piyush01123/Daily-Coding-Problems
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsol.py
45 lines (41 loc) · 1.08 KB
/
sol.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
42
43
44
45
def binarySearch(arr, key):
low=0
high=len(arr)-1
while high>=low:
mid = (low+high)//2
if arr[mid]>key:
high = mid-1
elif arr[mid]<key:
low = mid+1
elif arr[mid]==key:
return mid
return -1
def findPivot(arr):
low=0
high=len(arr)-1
while high>=low:
mid = (low+high)//2
# print('L=', low, 'M=', mid, 'H=', high)
if arr[mid]>arr[low] and arr[mid]<arr[high]:
return -1
if low==mid or high==mid:
# print(low, high, mid, 'returning')
return mid
elif arr[mid]<arr[low]:
high = mid
elif arr[mid]>arr[high]:
low = mid
def findIndex(arr, key):
pivot = findPivot(arr)
if pivot == -1:
return binarySearch(arr, key)
else:
if key>arr[0]:
return binarySearch(arr[:pivot+1], key)
elif key<arr[0]:
return pivot + binarySearch(arr[pivot:], key)
else:
return 0
if __name__=='__main__':
A=[5, 7, 8, 10, 2, 3]
print(findIndex(A, 2))