TreeMap.hpp

#pragma once
#include <map>

namespace samchon
{
    template <typename _Kty, typename _Ty, typename _Pr = std::less<_Kty>, typename _Alloc = std::allocator<std::pair<const _Kty, _Ty>>>
    class TreeMap
        : public std::map<_Kty, _Ty, _Pr, _Alloc>
    {
    private:
        typedef std::map<_Kty, _Ty, _Pr, _Alloc> super;

        //PROHIBIT ACCESS TO AT
        auto at(const _Kty &) -> _Ty& = delete;
        auto at(const _Kty &) const -> const _Ty& = delete;

    public:
        using super::super;

        auto has(const _Kty &key) const -> bool
        {
            return count(key) != 0;
        };

        auto get(const _Kty &key) -> _Ty&
        {
            return find(key)->second;
        };
        auto get(const _Kty &key) const -> const _Ty&
        {
            return find(key)->second;
        };

        void set(const _Kty &key, const _Ty &val)
        {
            iterator it = find(key);
            if (it != end())
                erase(it);

            insert({ key, val });
        };
        void set(const _Kty &key, const _Ty &&val)
        {
            iterator it = find(key);
            if (it != end())
                erase(it);
                
            insert({ key, val });
        };

        auto pop(const _Kty &key) -> _Ty
        {
            iterator it = find(key);
            _Ty val = it->second;

            erase(it);

            return val;
        };
    };
};