map重写比较器

最新推荐文章于 2024-03-08 10:33:16 发布

weixin_34014555

最新推荐文章于 2024-03-08 10:33:16 发布

阅读量323

点赞数

文章标签： c/c++ 数据结构与算法

本文详细介绍了如何在C++中为结构体重载比较运算符（< 和 >），以便将结构体用作map的key或放入set中。通过实例展示了如何实现并使用这些自定义比较器，确保数据结构按照特定规则排序。

结构体作为map的key或放入set中，需要重载<运算符，如下：

typedef struct tagRoadKey
{
    int m_i32Type;
    int m_i32Scale;

    bool operator <(const tagRoadKey& other) const // 注意是const函数！！
    {
        if (m_i32Type != other.m_i32Type) // 类型按升序排序
        {
            return (m_i32Type < other.m_i32Type);
        }
        else // 如果类型相同，按比例尺升序排序
        {
            return (m_i32Scale < other.m_i32Scale);
        }
    }

} RoadKey;

也可以重载>运算符，示例如下：

#include <iostream>
#include <string>
#include <map>

using namespace std;

class Array
{
private:
    int m_i32Num1;
    int m_i32Num2;

public:
    Array(int i32Num1, int i32Num2);
    bool operator >(const Array& other) const;
};

Array::Array(int i32Num1, int i32Num2)
{
    m_i32Num1 = i32Num1;
    m_i32Num2 = i32Num2;
}

bool Array::operator >(const Array& other) const
{
    if (m_i32Num1 > other.m_i32Num1)
    {
        return true;
    }
    else
    {
        return false;
    }
}

// 此结构体作为map的value
struct TInfo
{
    int m_i32Num1;
    int m_i32Num2;
};

int main(int argc, char* argv[])
{
    map<Array, TInfo, greater<Array> > stMap;

    TInfo stInfo1 = { 1, 1};
    stMap.insert(pair<Array, TInfo>(Array(1, 2), stInfo1));

    TInfo stInfo2 = { 2, 1, 1 };
    stMap.insert(pair<Array, TInfo>(Array(2, 2), stInfo2));

    TInfo stInfo3 = { 3, 1, 1 };
    stMap.insert(pair<Array, TInfo>(Array(3, 2), stInfo3));

    for (map<Array, TInfo, greater<Array> >::iterator it = stMap.begin(); it != stMap.end(); ++it)
    {
        cout << it->second.m_i32Num1 << endl;
    }

    return 0;
}
说明：
 map缺省是用less<Key>作为比较器，所以它要求作为Key的类要重载“<”操作符，没有重载“<”操作符，而是重载了“>”操作符就会报错。
反之，也可以显式地用greater<Key>作为比较器，此时就必要重载Key类中的“>”操作符了。
附：stl中map和set的声明，二者比较像，底层都是用红黑树实现的

template < class Key, class Compare = less<Key>,
           class Allocator = allocator<Key> > class set;

template < class Key, class T, class Compare = less<Key>,
           class Allocator = allocator<pair<const Key,T> > > class map;
 
template < class Key, class Compare = less<Key>,
           class Allocator = allocator<Key> > class multiset;
 
template < class Key, class T, class Compare = less<Key>,
           class Allocator = allocator<pair<const Key,T> > > class multimap;
 
从上面的声明可以看出，也可以定义一个函数对象Compare，声明map或set类型时传进入，如：

struct TTimeCompare
{
    bool operator ()(const CTimerEvent* po1, const CTimerEvent* po2)const
    {        
        return (po1->m_oNextTick < po2->m_oNextTick);
    }
};

typedef multiset<CTimerEvent*, TTimeCompare> TEventSet;


struct ltstr // less than
{
    bool operator()(const char* s1, const char* s2) const
    {
        return strcmp(s1, s2) < 0;
    }
};




set<const char*, ltstr> stSet; // set<Key, Compare, Alloc>
map<const char*, int, ltstr> stMap; // map<Key, Data, Compare, Alloc>




struct eqstr // equal
{
    bool operator()(const char* s1, const char* s2) const
    {
        return strcmp(s1, s2) == 0;
    }
};


hash_map<const char*, int, hash<const char*>, eqstr> stHashMap;  // hash_map<Key, Data, HashFcn, EqualKey, Alloc>


// 自定义hash函数
namespace std
{
    template<>
    struct hash<KEY_TYPE>
    {
        size_t operator()(const KEY_TYPE& key) const
        {
            //return key.Hash();
        }
    };
}

相等的时候返回false，否则报错