[C++] 간단한 람다식 정렬

요즘 간단하게나마 람다식을 짜는 것에 재미들렸다. 물론 편리하기도 하다!

 

vector<pair<int, int>> v;

 

v.push_back(make_pair(10, 10));

v.push_back(make_pair(2, 9));

v.push_back(make_pair(3, 8));

v.push_back(make_pair(4, 7));

v.push_back(make_pair(5, 6));

 

sort를 이용한 기본 정렬은 첫번째 원소 (first)를 기준으로 정렬해준다.

 

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

벡터 출력 결과

---

v의 첫번째 원소(v[i].first)가 큰 순서대로 (내림차순) 정렬하고 싶다면

sort(v.begin(), v.end(), [](pair<int, int> a, pair<int, int> b) {

        return a.first > b.first;

    });

벡터 출력 결과

 

---

v의 두번째 원소가 작은 순서대로 (오름차순) 정렬하고 싶다면

 

sort(v.begin(), v.end(), [](pair<int, int> a, pair<int, int> b) {

        return a.second < b.second;

    });

return 부분만 위와 같이 변경해주면 된다!

벡터 출력 결과

 

---

첫번째 원소의 내림차순, 첫번째 원소가 같다면 두번째 원소의 내림차순으로 정렬하기

 

vector 재정의

    v.push_back(make_pair(1, 10));

    v.push_back(make_pair(1, 9));

    v.push_back(make_pair(1, 8));

    v.push_back(make_pair(3, 7));

    v.push_back(make_pair(5, 6));

 

sort(v.begin(), v.end(), [](pair<int, int> a, pair<int, int> b) {

        if(a.first == b.first)

            return a.second > b.second;

        return a.first > b.first;

    });

 

벡터 출력 결과

 

다음엔 좀 더 심화된 람다식 응용을 공부해야겠다!

 

[출처] 내 머리