Middle+
Какво трябва да направите, за да използвате вашия клас като ключ в map или unordered_map?
sobes.tech AI
Отговор от 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 не може да хешира потребителски типове.
Начини за дефиниране на хешер:
-
Пренатоварване на
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; } -
Предаване на хешера като шаблонен параметър:
#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 също е необходим оператор за сравнение ==.