c++ map的一道key排序面试题引发的思考

问题描述: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;
    }
}

c++ map的一道key排序面试题引发的思考_第1张图片

输出结果:[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;
    }
}

c++ map的一道key排序面试题引发的思考_第2张图片

输出:[root@iZuf60vfno7b9n6pw24kdmZ admin]# ./main
wang 28
lia 30

你可能感兴趣的:(C++,c++)