您的位置:首页 > 其它

【map】【unordered_map】map和unordered_map中键类型为自定义类型的操作

2015-01-16 11:21 387 查看
STL中map的底层为红黑树,所以查找的时间复杂度为O(logn)

unordered_map是根据哈希值(遇到哈希值相同时用==号比较)寻找键,所以时间复杂度为O(1)

键类型为自定义类型时,map需要重载键类型的<符号,unordered_map需要定义键类型的哈希函数(在类外定义),以及重载键类型的==符号。

class person1
{  
public:  
    string name;  
    int age; 
    person1(string s,int i):name(s),age(i){}  
	//map必须重载键类型的<符号
	const bool operator<(const person1 &a)const
	{
		if(age!=a.age)
			return age<a.age;
		else
			return name<a.name;
	}
};  
  
class person2
{  
public:  
    string name;  
    int age; 
    person2(string s,int i):name(s),age(i){}  

	//unordered_map必须重载键类型的==符号,才能在哈希值相等的情况下找到准确的键值对
	const bool operator==(const person2 &a)const
	{
		return age==a.age && name==a.name;
	}
}; 

//person2类的哈希函数
struct person2_hash
{
	//unordered_map必须定义键类型的哈希函数
	size_t operator()(const person2 &p)const  
    {  
        return hash<int>()(p.age)+hash<char>()(p.name[0]);  
    }  
};

//测试代码
//map操作
map<person1,string> map_person;
person1 p1("Tom",15);
person1 p2("Tim",22);
//map_person.emplace("Tim",22,"driver"); 该行语法错误
map_person[p1]="student";
map_person[p2]="driver";
cout<<map_person[p1]<<endl;//输出 student
cout<<map_person[p2]<<endl;//输出 driver

//unordered_map操作
unordered_map<person2,string,person2_hash> unordered_person;
person2 x1("Ton",14);//x1和x2的哈希值相等
person2 x2("Tim",14);
person2 x3("John",50);
unordered_person[x1]="I am Ton";
unordered_person[x2]="I am Tim";
unordered_person[x3]="I am John";
cout<<unordered_person[x1]<<endl;//输出  I am Ton
cout<<unordered_person[x2]<<endl;//输出  I am Tim
cout<<unordered_person[x3]<<endl;//输出  I am John

上述代码中,person1类重载了<符号(为map操作服务),根据重载函数,可以看出map操作将按照升序存储数据。若希望map数据按照降序排列,还是要重载<符号(而不是重载>符号!),只不过函数代码里把原本返回true的地方改成false,原本返回false的改成true即可。

map需要的<符号也可以在类外定义,如下所示:

class person1
{  
public:  
    string name;  
    int age; 
    person1(string s,int i):name(s),age(i){}  
	//map必须重载键类型的<符号
	/*const bool operator<(const person1 &a)const
	{
		if(age!=a.age)
			return age<a.age;
		else
			return name<a.name;
	}*/
};  

struct compare
{
const bool operator()(const person1 &a,const person1 &b)const
	{
		if(b.age!=a.age)
			return a.age<b.age;
		else
			return a.name<b.name;
	}
};

//测试代码
map<person1,string,compa> map_person;
	person1 p1("Tom",15);
	map_person[p1]="uuu";
	cout<<map_person[p1]<<endl;//输出 uuu
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: