#pragma once

#include <algorithm>
#include <list>
#include <vector>

template <typename K, typename V>
class HashTable {
public:
    HashTable(size_t size)
        : size { size }
        , table(size)
    {
    }

private:
    size_t size;

    std::vector<std::list<std::pair<K, V>>> table;

    std::hash<K> hash_function;

    auto bucket_find_key(std::list<std::pair<K, V>> list, K key)
    {
        return std::find_if(list.begin(), list.end(), [&key](const std::pair<K, V>& pair) {
            return pair.first == key;
        });
    }

    bool bucket_contains_key(std::list<std::pair<K, V>> list, K key)
    {
        return bucket_find_key(list, key) != list.end();
    }

    std::list<std::pair<K, V>> get_bucket(K key)
    {
        size_t index = hash_function(key) % size;
        return table.at(index);
    }
};