STL矢量排序

时间:2015-09-08 07:40:45

标签: c++ stl

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

struct point{
    int index;
    int x;
    int y;
};

在x的基础上进行比较并在y !!的基础上打破关系

bool comp1(const point& p1,const point& p2){
    if(p1.x<p2.x)
        return true;
    if(p1.x==p2.x){
        if(p1.y<p2.y)
            return true;
    return false;
    }

    return true;
}

在y的基础上进行比较并在x !!的基础上打破关系

bool comp2(const point &p1,const point &p2){
    if(p1.y<p2.y)
        return true;
    if(p1.y==p2.y){
        if(p1.x<p2.x)
            return true;
    }
    return false;
}

int main() {
        int n;
        vector<point> px,py;
        cin>>n;
        point p;
        for(int i=0;i<n;i++){
            cin>>p.x>>p.y;
            p.index = i+1;
            px.push_back(p);
            py.push_back(p);

        }

        sort(px.begin(),px.end(),comp1);
        sort(py.begin(),py.end(),comp2);

        vector<point>::iterator it;
        for(it=px.begin();it!=px.end();it++)
            cout<<it->x<<" "<<it->y<<"\t";
        cout<<endl;

        for(it=py.begin();it!=py.end();it++)
            cout<<it->x<<" "<<it->y<<"\t   ";
        cout<<endl;

    return 0;
}

上面的代码产生了模糊的结果。我试图在增加x而不是增加y的基础上对点进行排序,反之亦然。

1 个答案:

答案 0 :(得分:2)

您可以使用std::pair()的内置operator<来简化逻辑,它会进行字典比较:

bool comp1(const point& p1,const point& p2){
  return std::make_pair(p1.x, p1.y) < std::make_pair(p2.x, p2.y);
}
// swap x and y order for comp2
bool comp2(const point& p1,const point& p2){
  return std::make_pair(p1.y, p1.x) < std::make_pair(p2.y, p2.x);
}