前言
sort用于C++中,对给定区间所有元素排序,在算法题时经常会用到sort排序。本文将会介绍几个常用的sort自定义排序。
sort介绍
sort函数有三个参数sort(start,end,cmp),一般直接使用从小到大排序sort(nums.begin(), nums.end() [, less< data-type >]);或者从大到小排序sort(nums.begin(), nums.end(), grater< data-type >());
尤其注意sort的第一个和第二个参数是指针,所以传入的必须是地址,且第二个要传入最后一个元素的下一个地址。
自定义排序规则
通过cmp我们可以自定义排序按照一定的规则,比如学生成绩排序,若成绩相同按名字的字典序排序。因此就可以根据需要来写一个cmp函数作为排序的准则传入到sort()中。
我们以这些数据为例
创建这些数据
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
class Student
{
public:
Student() {}
Student(string name, int score)
{
m_Name = name;
m_Score = score;
}
string m_Name;
int m_Score;
};
int main()
{
vector<Student> student;
Student s1("张三", 76);
Student s2("李四", 83);
Student s3("王五", 99);
Student s4("赵六", 78);
Student s5("九二", 76);
student.push_back(s1);
student.push_back(s2);
student.push_back(s3);
student.push_back(s4);
student.push_back(s5);
for (int i = 0; i < 5; i++)
cout << student[i].m_Name << "\t" << student[i].m_Score << endl;
return 0;
}
方式一:直接写bool类型cmp函数
bool cmp(Student s1, Student s2) //直接写bool类型cmp函数
{
if (s1.m_Score == s2.m_Score) return s1.m_Name < s2.m_Name; //若分数相同,按字母升序
return s1.m_Score > s2.m_Score; //成绩降序
}
sort(student.begin(), student.end(), cmp);
这里虽然使用了类,结构体同样可以
方式二:重载运算符> 或 <
class Student
{
public:
Student() {}
Student(string name, int score)
{
m_Name = name;
m_Score = score;
}
string m_Name;
int m_Score;
bool operator <(const Student& s2) const //重载<运算符,使名字升序
{
return m_Name < s2.m_Name;
}
bool operator >(const Student& s2) const //重载>运算符,使分数降序
{
return m_Score > s2.m_Score;
}
};
bool cmp(Student s1, Student s2)
{
if (s1.m_Score == s2.m_Score) return s1 < s2; // 直接使用重载运算符作比较,结果为分数相同,按名字字母升序
return s1 > s2; //按分数降序
}
sort(student.begin(), student.end(), cmp);
结果和方式一相同,注意重载写在类里面。如果使用sort(student.begin(), student.end(), greater< Student >())
将会按照">"的准则排序,及使分数降序;同理使用less<Student>()
则按照“<"的准则排序。
方式三:在sort的cmp函数直接写规则
语法为:sort(begin(), end(), [&](const data-type& a, const data-type& b) {return 规则语句} );
sort(student.begin(), student.end(), [&](const Student& s1, const Student& s2) {
if (s1.m_Score == s2.m_Score) return s1.m_Name < s2.m_Name;
return s1.m_Score > s2.m_Score;} );
结果还是同上,注意return一定要有。
[LC]1636. 按照频率将数组升序排序
本题官解就用到方式三进行排序,大家可以去试一下。