clean up of bsearch

This commit is contained in:
2025-10-27 00:20:36 +00:00
parent 8dbedb691d
commit 449d15c1c7

View File

@@ -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;
} }