mandlm/VectorSearch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: clean up

Browse source
This commit is contained in:
Michael Mandl 2024-03-22 10:59:35 +01:00
parent f080d00697
commit 66a54d85cc
Signed by: mandlm
GPG key ID: 4AA25D647AA54CC7
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lib_vector_search/include/bucket_finder.h
View file

@ -7,7 +7,7 @@
class Bucket {
private:
std::map<const char, WordRefList> groups_;
std::map<char, WordRefList> directory_;
public:
void insert(const WordList &word_list, size_t first_index, size_t last_index);

10
lib_vector_search/src/bucket_finder.cpp
View file

@ -6,20 +6,20 @@
void Bucket::insert(const WordList &word_list, size_t first_index,
size_t last_index) {
for (size_t index = first_index; index < last_index; ++index) {
for (auto index = first_index; index < last_index; ++index) {
const auto &current_word = word_list[index];
groups_[current_word.front()].push_back(&current_word);
directory_[current_word.front()].push_back(&current_word);
}
}
WordRefList Bucket::find_prefix(std::string_view search_term) const {
auto group_it = groups_.find(search_term.front());
if (group_it == groups_.cend()) {
auto directory_it = directory_.find(search_term.front());
if (directory_it == directory_.cend()) {
return {};
}
WordRefList result;
for (const auto *word : group_it->second) {
for (const auto *word : directory_it->second) {
if (word->starts_with(search_term)) {
result.push_back(word);
}