实战经验:自定义函数在STL比较函数sort中的使用

实战经验:自定义函数在STL比较函数sort中的使用

作者:BlogUpdater |  时间:2017-04-19 |  浏览:1318 |  评论已关闭 条评论

STL库中有自带的排序函数族,今天就来讲讲其中比较常用的sort函数。

该函数可以对容器进行排序,比如如下的代码:

vector<int> vec;
vec.push_back(3);
vec.push_back(5);
vec.push_back(6);
vec.push_back(1);
sort(vec.begin(), vec.end());

经过上述sort调用之后,容器中的数据内容为升序排列:1, 3, 5, 6

我们可以定义比较函数来制定排序规则,具体代码如下:

降序排列(从大到小):

sort(vec.begin(), vec.end(), [](int a, int b)->bool
{
    return a > b;
});

升序排列(从小到大):

sort(vec.begin(), vec.end(), [](int a, int b)->bool
{
    return a < b;
});

在上面的例子代码中,出现了第三个实参,这个实参是一个用户自定义的比较函数。该函数返回为bool类型,参数为两个元素(这里是int类型)。我们可以这样理解:如果这个函数返回true,则排序时会将第一个元素放在第二个元素之前。在降序例子中,我们指定比较条件为a > b,也即如果a的值大于b的值,比较函数返回true,那么第一个元素a会放在第二个元素b之前,也即:较大的那个数字在前,所以排序后为降序排列。升序的例子与之相反,大家可以自行理解。

下面我们给出一个求两个vector之间的差集的例子:

vector<int> vec1;
vec1.push_back(3);
vec1.push_back(5);
vec1.push_back(6);
vec1.push_back(1);

vector<int> vec2;
vec2.push_back(4);
vec2.push_back(1);
vec2.push_back(3);
vec2.push_back(7);

vector<int> vecDiff;
sort(vec1.begin(), vec1.end(), [](int a, int b)->bool
{
    return a < b;
});
sort(vec2.begin(), vec2.end(), [](int a, int b)->bool
{
    return a < b;
});
set_difference(vec1.begin(), vec1.end(), vec2.begin(), vec2.end(), std::back_inserter(vecDiff),
[](int a, int b)->bool 
{
    return a < b;
});

以上的例子运用了自定义的比较函数,使用set_difference获取了vec1和vec2的差集,具体来说,它比较了vec1和vec2的数据集,找出了vec1中相对vec2中的差异集合(即vec1中有,但是vec2中没有那些数据)。特别需要注意,在使用set_difference这一函数之前,必须确保两个容器时已经排序过的。

vec1:{1, 3, 5, 6}

vec2:{1, 3, 4, 7}

vecDiff: {5, 6}

对于容器中是自定义数据类型的情况,也可以使用上述方法来自定义排序规则。比如以下的伪代码:

vector<MyType> vec;
sort(vec.begin(), vec.end(), [](MyType a, MyType b)->bool
{
    return a.property > b.property;
});

另外需要注意的是,使用上述排序算法,需要包含以下头文件:

#include <vector>
#include <algorithm>
#include <iterator>
using namespace std;
标签:

评论已关闭。