[C++] Priority Queue & Pair 사용법

Priortity Queue와 Pair를 함께 사용하면 어떤 기능을 할 수 있는지 알아보자.

 

① 기본적인 사용
첫 번째 인자 기준으로 내림차순 정렬,
같다면 두 번째 인자를 기준으로 내림차순 정렬.

 

c++

닫기 줄바꿈

 priority_queue<pair<int , int>> pq;
    
    pq.push({1,2});
    pq.push({1,3});
    pq.push({2,2});
    pq.push({2,5});
    pq.push({3,1});

    while(!pq.empty()){
        cout << pq.top().first << pq.top().second << endl;
        pq.pop();
    }

 

② greater함수 사용
첫 번째 인자 기준으로 오름차순 정렬,
같다면 두 번째 인자 기준으로 오름차순 정렬.

 

c++

닫기 줄바꿈

priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> pq;
    
    pq.push({1,2});
    pq.push({1,3});
    pq.push({2,2});
    pq.push({2,5});
    pq.push({3,1});

    while(!pq.empty()){
        cout << pq.top().first << pq.top().second << endl;
        pq.pop();
    }

 

 

반응형

③ compare 함수 사용
원하는 대로 설정 할 수 있다.
첫 번째 인자 기준으로 내림차순 정렬,
같다면 두 번째 인자 기준으로 오름차순 정렬을 만들어 보았다.

c++

닫기 줄바꿈

struct cmp{
bool operator()(pair<int, int>&a, pair<int, int>&b) {
              if (a.first == b.first) {
                  return a.second > b.second;
              }
              else {
                  return a.first < b.first;
              }
          }
};

priority_queue<pair<int,int>,vector<pair<int,int>>,cmp> pq;

 

 

Pair에 대한 문법

2022.02.10 - [c++/문법] - [C++] Pair 와 Tuple

[C++] Pair 와 Tuple

 

Priority Queue 정리

2022.02.12 - [c++/data_structure_구현] - [c++][자료구조] heap 구현 / STL / Priority Queue 총 정리

[c++][자료구조] heap 구현 / STL / Priority Queue 총 정리