Назад к вопросам
Middle+
83
questionbank

Что нужно сделать, чтобы использовать свой класс в качестве ключа в map или unordered_map?

Sobes Copilot

Получайте ответы в реальном времени

Ответ от нейросети

sobes.tech AI

Для std::map:

Необходимо определить оператор сравнения < для вашего класса.

#include <map>
#include <string>

struct MyKey {
    int id;
    std::string name;

    // Определяем оператор "<"
    bool operator<(const MyKey& other) const {
        if (id != other.id) {
            return id < other.id;
        }
        return name < other.name;
    }
};

int main() {
    std::map<MyKey, int> myMap;
    MyKey key1{1, "apple"};
    MyKey key2{2, "banana"};
    myMap[key1] = 10;
    myMap[key2] = 20;
    return 0;
}

Для std::unordered_map:

Необходимо определить функцию хеширования (хешер) для вашего класса и, если необходимо, оператор сравнения на равенство ==. Стандартный хешер std::hash не умеет хешировать пользовательские типы.

Способы определения хешера:

  1. Перегрузка std::hash в пространстве имен std (не рекомендуется):

    #include <unordered_map>
    #include <string>
    #include <functional> // Для std::hash
    
    struct MyKey {
        int id;
        std::string name;
    
        // Определяем оператор "=="
        bool operator==(const MyKey& other) const {
            return id == other.id && name == other.name;
        }
    };
    
    // Специализация std::hash для MyKey
    namespace std {
        template <>
        struct hash<MyKey> {
            size_t operator()(const MyKey& key) const {
                // Пример простого комбинирования хешей
                // Более сложные и устойчивые хеш-функции могут быть предпочтительней
                return hash<int>()(key.id) ^ (hash<std::string>()(key.name) << 1);
            }
        };
    }
    
    int main() {
        std::unordered_map<MyKey, int> myUnorderedMap;
        MyKey key1{1, "apple"};
        MyKey key2{2, "banana"};
        myUnorderedMap[key1] = 10;
        myUnorderedMap[key2] = 20;
        return 0;
    }
    
  2. Передача хешера как шаблонного параметра:

    #include <unordered_map>
    #include <string>
    #include <functional>
    
    struct MyKey {
        int id;
        std::string name;
    
        // Определяем оператор "=="
        bool operator==(const MyKey& other) const {
            return id == other.id && name == other.name;
        }
    };
    
    // Определение структуры хешера
    struct MyKeyHash {
        size_t operator()(const MyKey& key) const {
            return std::hash<int>()(key.id) ^ (std::hash<std::string>()(key.name) << 1);
        }
    };
    
    int main() {
        // Передаем MyKeyHash как третий шаблонный параметр
        std::unordered_map<MyKey, int, MyKeyHash> myUnorderedMap;
        MyKey key1{1, "apple"};
        MyKey key2{2, "banana"};
        myUnorderedMap[key1] = 10;
        myUnorderedMap[key2] = 20;
        return 0;
    }
    

В обоих случаях для std::unordered_map также требуется оператор сравнения на равенство ==.