-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path34. Find First and Last Position of Element in Sorted Array.cpp
70 lines (67 loc) · 1.88 KB
/
34. Find First and Last Position of Element in 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
class Solution {
public:
// using std binary search : 19 ms
vector<int> searchRange1(vector<int>& nums, int target) {
auto it = upper_bound(nums.begin(),nums.end(),target-1);
if(it== nums.end() || (*it!=target))
return {-1,-1};
auto it1 = lower_bound(nums.begin(),nums.end(),target+1);
int start = distance(nums.begin(),it);
int end = distance(nums.begin(),it1);
return {start,end-1};
}
// binary search self; 23 ms
vector<int> searchRange2(vector<int>& nums, int target) {
int n = nums.size();
if(n==0)
return {-1,-1};
int l=0,r=n;
while(l<r){
int mid = l+(r-l)/2;
if(nums[mid]<target)
l=mid+1;
else r = mid;
}
//++l;
//cout<<"c";
if(l==n || nums[l]!=target)
return {-1,-1};
int start = l;
l=0,r=n;
while(l<r){
int mid = l+(r-l)/2;
if(nums[mid]<= target)
l=mid+1;
else r = mid;
}
--l;
return {start,l};
}
// one pass : 3 ms
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> result(2,-1);
int len = nums.size();
if(len==0)
return result;
int begin = 0;
int end = len-1;
while(nums[begin]< nums[end])
{
int mid = begin + ((end-begin)/2);
if(nums[mid]> target)
end = mid-1;
else if(nums[mid]< target)
begin = mid+1;
else
{
if(nums[begin]<target)
++begin;
if(nums[end]>target)
--end;
}
}
if(nums[begin]==target)
result[0]=begin,result[1]=end;
return result;
}
};