您的位置:首页 > 其它

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

2011-11-25 14:29 411 查看
结构体作为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();

}

};

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐