首页 > > 程序设计 > C/C++ >

C++对一组pair数据进行排序(sort函数的使用）

2018-06-17 21:01:23来源：未知阅读 ()

最近在写一个算法的时候，把一些数据存在了pair中，并且需要根据pair中first或者second的值对这些数据进行排序。比如：输入数据（1,2）、（4,2）、（3,3）、（2,1）根据first的值大小进行升序排序，输出（1,2）、（2,1）、（3,3）、（4,2）。
经过思索之后得到的实现方法如下：
首先将这些数据存在vector数组中，vector<pair<int,int>>vec;
然后使用sort函数对数组进行排序，这里就涉及到了sort函数的使用了。
下面是sort函数使用方法
函数声明：

template <class RandomAccessIterator>
  void stable_sort ( RandomAccessIterator first, RandomAccessIterator last );

template <class RandomAccessIterator, class Compare>
  void stable_sort ( RandomAccessIterator first, RandomAccessIterator last,
                     Compare comp );

参数解释：
（1）第一个是要排序的数组的起始地址。
（2）第二个是结束的地址（最后一位要排序的地址的下一地址）。
（3）第三个参数是排序的方法，可以是从大到小也可是从小到大，还可以不写第三个参数，此时默认的排序方法是从小到大排序。
我们可以根据自己的需求自定义第三个参数cmp函数，比如若要对整型数组降序排序，则可以这样来写cmp函数：

bool cmp（int a, int b）
{
    return a>b;//升序则为a<b
}

这个cmp参数使用起来非常方便，因为它正好能解决我们的pair排序问题。
我们只要根据数组中的元素类型定义cmp函数的参数，并且根据需求定义函数体

bool cmp（pair<int, int>a, pair<int, int>b）
{
    return a.first<b.first;//根据fisrt的值升序排序
    //return a.second<b.second;//根据second的值升序排序
}

然后调用sort函数sort（vec.begin(),vec.end(),cmp）。
便可以根据pair中first的值进行升序排序。

下面是具体代码实现

#include "stdafx.h"
#include <iostream>
#include <vector>
#include<algorithm>

using namespace std;

//根据first的值升序排序
bool cmp1(pair<int,int>a,pair<int,int>b)
{
    return a.first < b.first;
}

//根据second的值升序排序
bool cmp2(pair<int, int>a, pair<int, int>b)
{
    return a.second < b.second;
}
int main()
{
    vector<pair<int, int>>vec;
    vec.push_back({ 1,2 });
    vec.push_back({ 4,2 });
    vec.push_back({ 3,3 });
    vec.push_back({ 2,1 });
    sort(vec.begin(), vec.end(), cmp1);
    cout << "根据first的值升序排序:" << endl;
    for (auto it = vec.begin();it != vec.end();it++)
    {
        cout << "(" << it->first << "," << it->second << ")" << endl;
    }
    sort(vec.begin(), vec.end(), cmp2);
    cout << "根据second的值升序排序:" << endl;
    for (auto it = vec.begin();it != vec.end();it++)
    {
        cout << "(" << it->first << "," << it->second << ")" << endl;
    }
}