Trie.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_FACTORED_TRIE_HEADER_FILE
#define AI_TOOLBOX_FACTORED_TRIE_HEADER_FILE
 
#include <AIToolbox/Factored/Types.hpp>
#include <AIToolbox/Utils/IndexMap.hpp>
 
namespace AIToolbox::Factored {
    class Trie {
        public:
            Trie(Factors F);
 
            Factors getF() const;
 
            void reserve(size_t size);
 
            size_t insert(const PartialFactors & pf);
 
            size_t size() const;
 
            std::vector<size_t> filter(const Factors & f, size_t offset = 0) const;
 
            std::vector<size_t> filter(const PartialFactors & pf) const;
 
            std::vector<size_t> refine(const std::vector<size_t> & ids, const PartialFactors & pf) const;
 
            void erase(size_t id);
 
            void erase(size_t id, const PartialFactors & pf);
 
            const Factors & getFactors() const;
 
        private:
            std::vector<size_t> getAllIds() const;
 
            Factors F;
            size_t counter_;
 
            std::vector<std::vector<std::vector<size_t>>> ids_;
    };
}
 
#endif