第一种方式,编写比较函数并传入实参
用来组织一个容器中元素的操作的类型也是该容器类型的一部分。为了指定使用自定义的操作,必须在定义关联容器类型时提供此操作的类型。自定义的操作类型必须在尖括号中紧跟着元素类型给出。
在尖括号中出现的每个类型,就仅仅是一个类型而已。当我们创建一个容器时,才会以构造函数参数的形式提供真正的比较操作 (其类型必须与在尖括号类指定的类型相吻合) …… ①。
假设我们有类 x:
struct x {
int a,b;
x() = default;
x(int a1,int b1):a(a1), b(b1) { }
};
以及 comp 函数:
bool comp(const x &a, const x &b) {
if(a.a != b.a) return a.a < b.a;
return a.b < b.b;
}
如果我们想把这个自定义的操作用在 set 或者 map 等有序关联容器上,就应该这样:
map<x, int, decltype(cmp)*> mp(&comp);
multiset<x, decltype(cmp)*>mulse(&comp);
!注意: 尖括号内给出的自定义操作类型必须是函数指针且可以指向 comp 函数,然后定义完成时,在构造函数里面提供 comp (原因见上 语句①)。当然,用 &comp 和 comp(用 comp 在需要的时候会自动转换为一个指针) 。
第二种方式,重载函数调用运算符
可以自定义一个函数对象类,并重载函数调用运算符 “( )”。例如:
struct A {
int x,y;
};
struct comp {
bool operator() (const A& a,const A& b) const {
if(a.x != b.x) return a.x < b.x;
return a.y < b.y;
}
};
然后我们声明的时候同样需要在尖括号内带上第三个参数,表示一个函数对象类:
map<A,int,comp>mp;
测试Code:
#include<bits/stdc++.h>
using namespace std;
struct A {
int x,y;
};
struct comp {
bool operator() (const A& a,const A& b) const {
if(a.x != b.x) return a.x < b.x;
return a.y < b.y;
}
};
int main() {
map<A,int,comp>mp;
mp.insert({{2,1},1});
mp.insert({{1,2},2});
for(auto it : mp) cout << it.first.x << " " << it.first.y << " " << it.second << endl;
return 0;
}