Min stl priority_queue를 어떻게 만들 수 있습니까?


110

기본 stl 우선 순위 큐는 최대 1입니다 (Top 함수는 가장 큰 요소를 반환합니다).

단순화를 위해 int 값의 우선 순위 큐라고 가정하십시오.

답변:


189

사용 std::greater비교 함수로 :

std::priority_queue<int, std::vector<int>, std::greater<int> > my_min_heap;

4
@eriks 몇 가지 옵션이 있습니다. 클래스 정의 중 하나는 operator>매력처럼 작동 std::greater합니다. 원하는 std::greater경우 대신 자신의 펑터를 작성할 수도 있습니다.
AraK 2010 년

2
@AraK, 난 당신 말은 생각 operator<)
피터 알렉산더

15
std :: greater <int>를 사용하려면 #include <functional>
reggaeguitar

3
표준 템플릿 라이브러리의 @CarryonSmiling에서 vectordeque클래스는 기본 컨테이너가 priority_queue에 대해 충족해야하는 요구 사항을 충족합니다. 사용자 지정 컨테이너 클래스를 사용할 수도 있습니다. 당신은에 훨씬 더 정교한 설명 찾을 수 있습니다 cplusplus.com/reference/queue/priority_queue
Tanmay Garg를

3
누군가 min heap이 less <int>가 아닌 greater <int>를 사용하는 이유를 설명 할 수 있습니까?
Telenoobies

45

한 가지 방법은 우선 순위가 반전되도록 일반 우선 순위 대기열에서 작동 할 적절한 비교기를 정의하는 것입니다.

 #include <iostream>  
 #include <queue>  
 using namespace std;  

 struct compare  
 {  
   bool operator()(const int& l, const int& r)  
   {  
       return l > r;  
   }  
 };  

 int main()  
 {  
     priority_queue<int,vector<int>, compare > pq;  

     pq.push(3);  
     pq.push(5);  
     pq.push(1);  
     pq.push(8);  
     while ( !pq.empty() )  
     {  
         cout << pq.top() << endl;  
         pq.pop();  
     }  
     cin.get();  
 }

각각 1, 3, 5, 8을 출력합니다.

STL 및 Sedgewick 구현을 통해 우선 순위 대기열을 사용하는 몇 가지 예가 여기 에 나와 있습니다 .


1
최소 우선 순위 대기열을 구현하기 위해 l <r이 아닌 l> r을 사용하는 이유를 설명해 주시겠습니까?
Dhruv Mullick 2014 년

3
우선 순위 큐의 기본 비교기는 l <r입니다. 생성자 기본 매개 변수 에서 확인할 수 있습니다 . l> r 또는 r <l을 수행하면 그 반대가됩니다.
Diaa 2014

1
@AndyUK 안녕하세요, 왜 비교 연산자를 구현하기 위해 구조체를 사용합니까? 미리 감사드립니다
AER 2018

C ++의 기본 우선 순위 대기열은 최대 우선 순위 대기열입니다.
qwr

30

세 번째 템플릿 매개 변수 priority_queue는 비교 자입니다. 사용하도록 설정하십시오 greater.

예 :

std::priority_queue<int, std::vector<int>, std::greater<int> > max_queue;

당신은해야합니다 #include <functional>위해 std::greater.


@Potatoswatter : 항상 그런 것은 아닙니다.
전능하신 낙타 Moha

11
그것은 또한 사용법 #include <기능>에 언급 때문 허용 대답보다 낫다
reggaeguitar

22

여러 가지 방법으로 할 수 있습니다.
1. greater비교 기능으로 사용 :

 #include <bits/stdc++.h>

using namespace std;

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

    while(!pq.empty())
    {
        int r = pq.top();
        pq.pop();
        cout<<r<< " ";
    }
    return 0;
}

2. 부호를 변경하여 값을 삽입합니다 (양수에는 마이너스 (-) 사용, 음수에는 플러스 (+) 사용).

int main()
{
    priority_queue<int>pq2;
    pq2.push(-1); //for +1
    pq2.push(-2); //for +2
    pq2.push(-3); //for +3
    pq2.push(4);  //for -4

    while(!pq2.empty())
    {
        int r = pq2.top();
        pq2.pop();
        cout<<-r<<" ";
    }

    return 0;
}

3. 사용자 정의 구조 또는 클래스 사용 :

struct compare
{
    bool operator()(const int & a, const int & b)
    {
        return a>b;
    }
};

int main()
{

    priority_queue<int,vector<int>,compare> pq;
    pq.push(1);
    pq.push(2);
    pq.push(3);

    while(!pq.empty())
    {
        int r = pq.top();
        pq.pop();
        cout<<r<<" ";
    }

    return 0;
}

4. 사용자 정의 구조 또는 클래스를 사용하여 순서에 관계없이 priority_queue를 사용할 수 있습니다. 급여에 따라 내림차순으로 정렬하고 동률이면 연령에 따라 사람들을 정렬한다고 가정합니다.

    struct people
    {
        int age,salary;

    };
    struct compare{
    bool operator()(const people & a, const people & b)
        {
            if(a.salary==b.salary)
            {
                return a.age>b.age;
            }
            else
            {
                return a.salary>b.salary;
            }

    }
    };
    int main()
    {

        priority_queue<people,vector<people>,compare> pq;
        people person1,person2,person3;
        person1.salary=100;
        person1.age = 50;
        person2.salary=80;
        person2.age = 40;
        person3.salary = 100;
        person3.age=40;


        pq.push(person1);
        pq.push(person2);
        pq.push(person3);

        while(!pq.empty())
        {
            people r = pq.top();
            pq.pop();
            cout<<r.salary<<" "<<r.age<<endl;
    }
  1. 연산자 오버로딩으로 동일한 결과를 얻을 수 있습니다.

    struct people
    {
    int age,salary;
    
    bool operator< (const people & p)const
    {
        if(salary==p.salary)
        {
            return age>p.age;
        }
        else
        {
            return salary>p.salary;
        }
    }};

    주요 기능 :

    priority_queue<people> pq;
    people person1,person2,person3;
    person1.salary=100;
    person1.age = 50;
    person2.salary=80;
    person2.age = 40;
    person3.salary = 100;
    person3.age=40;
    
    
    pq.push(person1);
    pq.push(person2);
    pq.push(person3);
    
    while(!pq.empty())
    {
        people r = pq.top();
        pq.pop();
        cout<<r.salary<<" "<<r.age<<endl;
    }

당신은 의미하지 않는다 bool operator > (const people & p)const 5) 연산자 오버로딩
Rockstar5645

1
사실, 오른쪽, 5)이 내가 본 적 없어요, 단지 이상해, 작업을 수행 <가 과부하에 더 나은, 그렇게 과부하가 없습니다 >사용greater<people>
Rockstar5645

19

C ++ 11에서는 편의를 위해 별칭을 만들 수도 있습니다.

template<class T> using min_heap = priority_queue<T, std::vector<T>, std::greater<T>>;

다음과 같이 사용하십시오.

min_heap<int> my_heap;

8

이 문제를 해결하는 한 가지 방법은 priority_queue에서 각 요소의 음수를 푸시하여 가장 큰 요소가 가장 작은 요소가되도록하는 것입니다. 팝 작업을 할 때 각 요소의 부정을 취하십시오.

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

int main(){
    priority_queue<int> pq;
    int i;

// push the negative of each element in priority_queue, so the largest number will become the smallest number

    for (int i = 0; i < 5; i++)
    {
        cin>>j;
        pq.push(j*-1);
    }

    for (int i = 0; i < 5; i++)
    {
        cout<<(-1)*pq.top()<<endl;
        pq.pop();
    }
}

3

위의 모든 답변을 바탕으로 우선 순위 대기열을 만드는 방법에 대한 예제 코드를 만들었습니다. 참고 : C ++ 11 이상 컴파일러에서 작동합니다.

#include <iostream>
#include <vector>
#include <iomanip>
#include <queue>

using namespace std;

// template for prirority Q
template<class T> using min_heap = priority_queue<T, std::vector<T>, std::greater<T>>;
template<class T> using max_heap = priority_queue<T, std::vector<T>>;

const int RANGE = 1000;

vector<int> get_sample_data(int size);

int main(){
  int n;
  cout << "Enter number of elements N = " ; cin >> n;
  vector<int> dataset = get_sample_data(n);

  max_heap<int> max_pq;
  min_heap<int> min_pq;

  // Push data to Priority Queue
  for(int i: dataset){
    max_pq.push(i);
    min_pq.push(i);
  }

  while(!max_pq.empty() && !min_pq.empty()){
    cout << setw(10) << min_pq.top()<< " | " << max_pq.top() << endl;
    min_pq.pop();
    max_pq.pop();
  }

}


vector<int> get_sample_data(int size){
  srand(time(NULL));
  vector<int> dataset;
  for(int i=0; i<size; i++){
    dataset.push_back(rand()%RANGE);
  }
  return dataset;
}

위 코드의 출력

Enter number of elements N = 4

        33 | 535
        49 | 411
       411 | 49
       535 | 33

1

우리는 여러 가지 방법으로 이것을 할 수 있습니다.

템플릿 비교기 매개 변수 사용

    int main() 
    {
      priority_queue<int, vector<int>, greater<int> > pq;

      pq.push(40);
      pq.push(320);
      pq.push(42);
      pq.push(65);
      pq.push(12);

      cout<<pq.top()<<endl;
      return 0;
    }

사용 된 정의 된 compartor 클래스 사용

     struct comp
     {
        bool operator () (int lhs, int rhs)
        {
           return lhs > rhs;
        }
     };

    int main()
    {
       priority_queue<int, vector<int>, comp> pq;

       pq.push(40);
       pq.push(320);
       pq.push(42);
       pq.push(65);
       pq.push(12);

       cout<<pq.top()<<endl;

       return 0;
    }
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.