-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathdouble_bi_search.cpp
84 lines (83 loc) · 1.85 KB
/
double_bi_search.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
#include <vector>
#include <iostream>
int FindMax(const std::vector<int> &data, int start, int end)
{
int mid = (start + end) / 2;
int max = mid;
if (data[mid] > data[mid - 1] && data[mid] > data[mid + 1])
{
return mid;
}
else
{
if (data[mid] > data[mid + 1])
{
FindMax(data, start, mid);
}
else
{
FindMax(data, mid, end);
}
}
}
int BiSearch(const std::vector<int> &data, int start, int end, int target, bool up)
{
int mid = (start + end) / 2;
if (data[mid] == target)
{
return mid;
}
if (start == end)
{
return data.size();
}
if (up)
{
if (data[mid] > target)
{
BiSearch(data, start, mid, target, up);
}
else
{
BiSearch(data, mid, end, target, up);
}
} else {
if (data[mid] > target)
{
BiSearch(data, mid, end, target, up);
}
else
{
BiSearch(data, start, mid, target, up);
}
}
}
std::vector<int> DoubleBiSearch(const std::vector<int> &data, const int &target)
{
int max = FindMax(data, 0, data.size() - 1);
std::cout << "max:" << max << std::endl;
bool up = true;
int ret1 = BiSearch(data, 0, max, target, up);
int ret2 = BiSearch(data, max, data.size() - 1, target, !up);
std::vector<int> rets;
if (ret1 != data.size())
{
rets.push_back(ret1);
}
if (ret2 != data.size())
{
rets.push_back(ret2);
}
return rets;
}
int main(int argc, char const *argv[])
{
std::vector<int> data = {
1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1};
std::vector<int> found = DoubleBiSearch(data, 5);
for (auto ret : found)
{
std::cout << "found:" << ret << std::endl;
}
return 0;
}