sort:
struct node
{
int l, r;
//返回一个bool值(排序涉及是否交换),两个const防止调用的变量发生变化
bool operator <(const node& a)const {
return l < a.l;//小于号 代表 从小到大 排序
//return l != a.l ? l < a.l : r > a.r 也可完成更复杂的排序
//如果 l 不同就按 l 从小到大排序,否则按 r 从大到小排序
}
}z[N];
//另一种写法,cmp比较函数
bool cmp(node a, node b) {
return a.l < b.l;
}
结构体内部设定,重载设定好后就可以直接存入 z 数组然后sort排序
cin >> n;
for (int i = 0; i < n; i++)
cin >> z[i].l >> z[i].r;
sort(z, z + n);//sort(z, z + n, cmp)
for (int i = 0; i < n; i++)
cout << z[i].l << ' ' << z[i].r << endl;
优先队列 priority_queue:
可用自带的函数greater、less来设定小根堆、大根堆,但缺点是只能根据第一个关键字排序:
struct Ver
{
int pre, cur, id;
//只能根据 pre 来排序
};
int main() {
priority_queue<Ver, vector<Ver>, greater<Ver> > q1;
//存入堆的类型、存放的容器、排序类型
priority_queue<Ver, vector<Ver>, less<Ver> > q2;
return 0;
}
手写重载运算符bool operator:
struct Ver
{
int pre, cur, id;
bool operator < (const Ver& a)const {
// return id < a.id;
//注意堆默认状态下是从大到小排序(大根堆),与sort相反
//所以这里是反过来:根据 id 从大到小排序
//可以实现更复杂的排序,不只是根据一个关键字:
if (id != a.id)return id < a.id;
else if (cur != a.cur)return cur > a.cur;
else return pre < a.pre;
}
};
int main() {
priority_queue<Ver> q;
//定义也简单
q.push({ 1,1,1 });
while (q.size())
{
Ver t = q.top();
pri("%d %d %d\n", t.pre, t.cur, t.id);
sca("%d %d %d", &t.pre, &t.cur, &t.id);
q.push(t);
}
return 0;
}