首页 > 代码库 > 有序容器自主定义排序器

有序容器自主定义排序器

STL中的set和map是有序容器,使用时如果希望根据自己的需要来设定排序器,通常有一下两种方式。

1.如果容器中直接存储对象的话,那么我们可以在对象类中重载<即可,内置类型的话就不需要了,因为有默认的

2.如果存储的不是直接对象的话比如对象的指针(通常为智能指针),这个时候我们就要定义自己的比较器。而比较器的写法一般有两种。

 ->1.类内重载函数调用运算符的方法。

 ->2.以函数的方式提供比较器。


对于第一种方法是非常简单而且经常用的,这里不再赘述。

下面主要以一个简单的例子来介绍第二种方法中比较器的两种写法(这里为了方便说明,暂时存储对象)。

student.h

class Student{
public:
	Student(const string &sid) : id(sid){}

	void print()const{
		cout << id << endl;
	}

	string getId()const{
		return id;
	}

private:
	string id;
};


main.cpp

struct Compare{
	//override the operator ()
	bool operator()(const Student &ls, const Student &rs)const{
		return ls.getId() < rs.getId();
	}
};

bool compare(const Student &ls, const Student &rs){
	return ls.getId() < rs.getId();
}

int main(){
	/*the first type-----define a class as the comparator*/
	set<Student, Compare> ms;

	ms.insert(Student("222"));
	ms.insert(Student("111"));
	ms.insert(Student("333"));

	auto ite = ms.begin();

	for (; ite != ms.end(); ++ite){
		ite->print();
	}

	/*the second type----define a function as the comparator*/
	/*
	set<Student, bool (*)(const Student &ls, const Student &rs)> ms(compare);

	ms.insert(Student("222"));
	ms.insert(Student("111"));
	ms.insert(Student("333"));

	auto ite = ms.begin();

	for (; ite != ms.end(); ++ite){
		ite->print();
	}
	*/
	return 0;
}