[C++] pair vector를 sort! 두번째값을 기준으로도 정렬 가능하다거!

 vector<pair<int, int> > v1;
 vector<pair<string, int>> v2;
 vector<pair<int, double>> v3;

좌표, 이름이 따로 있는 숫자 등을 담을 때 자주 쓰는 pair로 이루어진 벡터를 정렬하고자 한다..! 

 

 

// test.cpp
#include <bits/stdc++.h>
using namespace std;

int main()
{
    vector<pair<int, int> > v;
    v.push_back(make_pair(1, 2));
    v.push_back(make_pair(1, 0));
    v.push_back(make_pair(2, 0));
    v.push_back(make_pair(2, 2));

    sort(v.begin(), v.end());
    for (int i = 0; i < 4; ++i)
    {
        cout << v[i].first << " " << v[i].second << "\n";
    }
}

위의 코드를 실행하면  

// output of test.cpp
1 0
1 2
2 0
2 2

이처럼 first 값을 기준으로 오름차순, first 값이 같을 때는 second 값을 기준으로 오름차순. 이 규칙을 가지고 정렬된다. 

 

근데!! 여기서 조금 더 깐깐한 기준을 넣고 싶다면..? 

 

예를 들어 first 값을 기준 오름차순으로 정렬하되, first 값이 같을 때는 second 값을 기준 내림차순으로 정렬하고 싶다면!?

 

 

해결방법은 간단하다! 

sort 함수에 세번째 인자로 자기가 원하는 기준을 리턴하는 boolean 함수를 넣어주면 된다.

 

위에서 언급한 그 기준을 직접 만들어보자. 

// modified_test.cpp
bool compare(const pair<int, int> &a, const pair<int, int> &b)
{
    if (a.first == b.first) // first 값이 같다면?
    {
        return a.second > b.second; second 값이 더 큰게 앞으로 가게 해라!
    }
    return a.first < b.first; // first 값이 같지 않다면 더 큰 값이 뒤로 가게 해라!
}

이러한 함수를 sort 함수 세번째 인자로 넣어주면 된다. 아래처럼 말이다!

sort(v.begin(), v.end(), compare);

 

 

이렇게 규칙을 넣어주면 아까의 실행 결과는 우리의 의도대로 아래와 같이 변하게 된다!

// output of modified_test.cpp
1 2
1 0
2 2
2 0

 

 

 

---

bool compare(const pair<int, int> &a, const pair<int, int> &b)
{
    if (a.first == b.first)
    {
        return a.second > b.second;
    }
    return a.first < b.first;
}

위에서 추가해줬던 함수를 더 와닿게 이해하려면 a를 정렬된 벡터 기준 앞의 원소, b를 뒤의 원소라고 가정하고 보면 될 것 같다.

우리의 의도대로 정렬된 벡터라면..

두 원소의 first 값이 같지 않다면 앞에 있는 원소(a)의 fisrt 값이 더 작을 것이다. 이는 위 코드에서 if 문에 해당하지 않으니 가장 아래의 return 문이 실행될 것이다. 이는 앞에 있는 원소(a)의 first 값이 뒤에 오는 원소(b)의 first 값보다 작게 해! 라는 것을 의미한다.

 

두 원소의 first 값이 같다면 앞에 있는 원소(a)의 second 값이 뒤에 오는 원소(b)의 second 값보다 작을 것이다. 이는 위 코드에서 if문에 해당하게 되고, 조건문 안의 return 문이 실행될 것이다. 이는 앞에 있는 원소(a)의 second 값이 뒤에 오는 원소(b)의 second 값보다 크게 해! 라는 것을 의미한다.