Middle+
O que é necessário fazer para usar a sua classe como chave em um map ou unordered_map?
sobes.tech IA
Resposta da IA
Para std::map:
É necessário definir o operador de comparação < para a sua classe.
#include <map>
#include <string>
struct MyKey {
int id;
std::string name;
// Definir o operador "<"
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;
}
Para std::unordered_map:
É necessário definir uma função de hash (hasher) para a sua classe e, se necessário, o operador de comparação de igualdade ==. O hash padrão std::hash não consegue fazer hash de tipos personalizados.
Formas de definir o hash:
-
Sobrecarga de
std::hashno espaço de nomesstd(não recomendado):#include <unordered_map> #include <string> #include <functional> // Para std::hash struct MyKey { int id; std::string name; // Definir o operador "==" bool operator==(const MyKey& other) const { return id == other.id && name == other.name; } }; // Especialização de std::hash para MyKey namespace std { template <> struct hash<MyKey> { size_t operator()(const MyKey& key) const { // Exemplo de combinação simples de hashes // Funções de hash mais complexas e robustas podem ser preferidas 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; } -
Passar o hash como parâmetro de modelo:
#include <unordered_map> #include <string> #include <functional> struct MyKey { int id; std::string name; // Definir o operador "==" bool operator==(const MyKey& other) const { return id == other.id && name == other.name; } }; // Definição da estrutura do hash struct MyKeyHash { size_t operator()(const MyKey& key) const { return std::hash<int>()(key.id) ^ (std::hash<std::string>()(key.name) << 1); } }; int main() { // Passar MyKeyHash como terceiro parâmetro de modelo std::unordered_map<MyKey, int, MyKeyHash> myUnorderedMap; MyKey key1{1, "apple"}; MyKey key2{2, "banana"}; myUnorderedMap[key1] = 10; myUnorderedMap[key2] = 20; return 0; }
Em ambos os casos, também é necessário o operador de comparação de igualdade ==.