#include <array>
#include <cstring>
#include <iostream>
#include <memory>
#include <queue>
#include <stdexcept>
#include <string>
#include <vector>

#ifndef WORDTREE_HPP
    #define WORDTREE_HPP

class WordTree
{
    struct TreeNode
    {
        bool m_endOfWord;
        std::array<std::shared_ptr<TreeNode>, 26> m_children;

        TreeNode();
    };
    std::shared_ptr<TreeNode> m_root;

  public:
    WordTree();
    ~WordTree();

    void add(const std::string& word);
    bool find(const std::string& word);
    std::vector<std::string> predict(std::string partial, std::uint8_t howMany);
    std::size_t size();
};

#endif // WORDTREE_HPP