forked from AnasImloul/Leetcode-Solutions
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSearch in Rotated Sorted Array.cpp
94 lines (92 loc) · 2.37 KB
/
Search in Rotated Sorted Array.cpp
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
class Solution
{
public:
int search(vector<int>& nums, int target)
{
int n=nums.size()-1;
if(nums[0]==target)
return 0;
if(nums.size()==1)
return -1;
// find the index of the minimum element ie pivot
int pivot=-1;
int low=0;
int high=n;
while(low<=high)
{
int mid=(low+high)/2;
if(mid==0)
{
if(nums[mid]< nums[mid+1] && nums[mid]< nums[n])
{
pivot=mid;
break;
}
else
{
low=mid+1;
}
}
else if(mid==n)
{
if(nums[mid] < nums[0] && nums[mid] < nums[mid-1])
{
pivot=mid;
break;
}
else
{
high=mid-1;
}
}
else if(nums[mid]<nums[mid+1] && nums[mid]<nums[mid-1])
{
pivot=mid;
break;
}
else if(nums[mid]>=nums[0] && nums[mid]<=nums[n])
high=mid-1;
else if(nums[mid] >= nums[0])
low=mid+1;
else if(nums[mid] < nums[0])
high=mid-1;
}
cout<<pivot<<endl;
if(target < nums[0])
{
int low1=pivot;
int high1=n;
while(low1<=high1)
{
int mid1=(low1+high1)/2;
if(nums[mid1]==target)
return mid1;
else if(nums[mid1] > target)
high1=mid1-1;
else if(nums[mid1]<target)
low1=mid1+1;
}
}
if(target > nums[0])
{
int low2=0;
int high2=pivot-1;
if(pivot==0)
{
low2=0;
high2=n;
}
while(low2<=high2)
{
int mid2=(low2+high2)/2;
if(nums[mid2]==target)
return mid2;
else if(nums[mid2] > target)
high2=mid2-1;
else if(nums[mid2]<target)
low2=mid2+1;
}
}
return -1;
}
};