vector 정렬하기

임의의 클래스에 대해 sort함수로 정렬할 기준을 세우는것은 operator를 overloading 하면 되지만,

원하는 순간마다 사용자가 원하는 정렬을 하기는 쉽지 않다.

그럴때는 아래의 방법을 사용하면 된다.

#include<iostream>
#include<vector>
#include<random>
#include<algorithm>
 
using namespace std;
 
class Node {
public:
    int a, b;
    Node(int a, int b): a(a), b(b){}
    Node():Node(-1,-1){}
 
    friend ostream& operator<<(ostream& os, const Node &node) {
        os << "a : " << node.a << ", b : " << node.b << endl;;
        return os;
    }
};
 
vector<Node> nodes_a, nodes_b;
 
bool CompA(const Node& lhs, const Node& rhs) {
    return lhs.a < rhs.a;
}
 
bool CompB(const Node& lhs, const Node& rhs) {
    return lhs.b < rhs.b;
}
 
int main(void) {
    for (int i = 0; i < 10; i++) {
        nodes_a.push_back(Node(rand() % 100, rand() % 100));
        nodes_b.push_back(Node(rand() % 100, rand() % 100));
    }
 
    sort(nodes_a.begin(), nodes_a.end(), CompA);
    sort(nodes_b.begin(), nodes_b.end(), CompB);
 
    cout << "case A" << endl;
    for (int i = 0; i < nodes_a.size(); i++)
        cout << nodes_a[i];
 
    cout << "\n\n";
 
    cout << "case A" << endl;
    for (int i = 0; i < nodes_b.size(); i++)
        cout << nodes_b[i];
 
    return 0;
}