问题描述:map里面的key可以是自定义的数据类型?如果是,该自定义数据类型有什么特点?
答案:key可以是自定义数据类型,需要重载operator<运算符,因为map实现的机制是红黑树,默认是根据key的ASCII进行排序,如果是自定义数据类型,需要指定排序规则
c++ 实现demo
class A{
public:
A(std::string name_): name(name_){}
//运算符重载
bool operator<(const A& a) const{
return this->name > a.name ? true : false;
}
//重载输出运算符,否则first元素无法输出
friend std::ostream& operator<<(std::ostream& output, const A& a);
private:
std::string name;
};
std::ostream& operator<<(std::ostream& output, const A& a){
output<<a.name;
return output;
}
void test_03(){
std::map<A,int> map_{{A("wang"), 28}, {A("lia"), 30}};
for(auto& element : map_){
std::cout<<element.first<<" "<<element.second<<std::endl;
}
}
输出结果:[root@iZuf60vfno7b9n6pw24kdmZ admin]# ./main
wang 28
lia 30
通过自定义谓词进行排序
通过谓词进行规则设定,然后排序,效果也是一样的。
class B{
public:
friend class Compare;
B(std::string name_): name(name_){}
friend std::ostream& operator<<(std::ostream& output, const B& a);
private:
std::string name;
};
std::ostream& operator<<(std::ostream& output, const B& a){
output<<a.name;
return output;
}
class Compare{
public:
bool operator()(const B& a1, const B& a2){
return a1.name > a2.name ? true : false;
}
};
void test_04(){
std::map<B,int, Compare> map_;
map_.insert(std::make_pair<B,int>(B("wang"), 28));
map_.insert(std::make_pair<B,int>(B("lia"), 30));
for(auto& element : map_){
std::cout<<element.first<<" "<<element.second<<std::endl;
}
}
输出:[root@iZuf60vfno7b9n6pw24kdmZ admin]# ./main
wang 28
lia 30