clean up of bsearch
This commit is contained in:
@@ -4,6 +4,7 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
|
#include <cassert>
|
||||||
#include <cstdlib>
|
#include <cstdlib>
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
@@ -17,9 +18,7 @@ using std::ostream;
|
|||||||
using std::string;
|
using std::string;
|
||||||
using std::vector;
|
using std::vector;
|
||||||
|
|
||||||
vector<int> arr;
|
ostream &print_arr(ostream &os, std::vector<int> &arr)
|
||||||
|
|
||||||
ostream &print_arr(ostream &os)
|
|
||||||
{
|
{
|
||||||
os << "[";
|
os << "[";
|
||||||
for (size_t i = 0; i < arr.size(); ++i)
|
for (size_t i = 0; i < arr.size(); ++i)
|
||||||
@@ -27,31 +26,46 @@ ostream &print_arr(ostream &os)
|
|||||||
return os << "]";
|
return os << "]";
|
||||||
}
|
}
|
||||||
|
|
||||||
int bsearch(int n, int l = 0, int u = arr.size() - 1)
|
int bsearch(int n, std::vector<int> arr)
|
||||||
{
|
{
|
||||||
int midpoint = ((u + l) / 2);
|
int l = 0;
|
||||||
if (l >= u || u <= 0)
|
int u = arr.size() - 1;
|
||||||
return -1;
|
while (l <= u)
|
||||||
int val = arr[midpoint];
|
{
|
||||||
if (val == n)
|
int midpoint = l + ((u - l) / 2);
|
||||||
return midpoint;
|
int val = arr[midpoint];
|
||||||
else if (val > n)
|
|
||||||
return bsearch(n, l, midpoint - 1);
|
if (val == n)
|
||||||
else
|
{
|
||||||
return bsearch(n, midpoint + 1, u);
|
return midpoint;
|
||||||
|
}
|
||||||
|
else if (val > n)
|
||||||
|
{
|
||||||
|
u = midpoint - 1;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
l = midpoint + 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int main(void)
|
int main(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
std::ifstream input("bsearch.txt");
|
std::ifstream input(argc > 1 ? argv[1] : "bsearch.txt");
|
||||||
string line;
|
std::vector<int> arr;
|
||||||
while (std::getline(input, line))
|
|
||||||
arr.push_back(std::stoi(line));
|
for (std::string line; std::getline(input, line);
|
||||||
|
arr.push_back(std::stoi(line)))
|
||||||
|
continue;
|
||||||
|
|
||||||
std::sort(std::begin(arr), std::end(arr));
|
std::sort(std::begin(arr), std::end(arr));
|
||||||
string inp;
|
|
||||||
cout << "Enter number to search: ";
|
for (size_t i = 0; i < arr.size(); ++i)
|
||||||
cin >> inp;
|
{
|
||||||
int to_search = std::stoi(inp);
|
int index = bsearch(arr[i], arr);
|
||||||
cout << bsearch(to_search) << endl;
|
assert(index == (int)i);
|
||||||
|
}
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user