web-dev-qa-db-ja.com

Min stl priority_queueを作成するにはどうすればよいですか？

デフォルトのstl優先度キューは最大1です（Top関数は最大の要素を返します）。

簡単にするために、int値の優先キューであるとします。

c++stlpriority-queue
95
amitlicht

比較関数としてstd::greaterを使用します。

std::priority_queue<int, std::vector<int>, std::greater<int> > my_min_heap;
172
James McNellis

1つの方法は、通常の優先度キューで動作する適切なコンパレータを定義して、優先度が逆になるようにすることです。

 #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の実装 を介して優先キューを使用するいくつかの例が here で与えられています。

41
AndyUK

priority_queueの3番目のテンプレートパラメータはコンパレータです。 greaterを使用するように設定します。

例えば.

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

#include <functional>にはstd::greaterが必要です。

29
Peter Alexander

複数の方法でそれを行うことができます：
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;
}
18
Taohidul Islam

C++ 11では、便宜上エイリアスを作成することもできます。

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

そして、次のように使用します：

min_heap<int> my_heap;
16
mechatroner

この問題を解決する1つの方法は、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();
    }
}
8
PAVAN BANSAL

上記のすべての回答に基づいて、優先度キューを作成する方法のサンプルコードを作成しました。 注：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
3
ajayramesh

これにはいくつかの方法があります。

テンプレートコンパレータパラメータの使用

    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;
    }

使用済みの定義済みコンパートメントクラスを使用する

     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;
    }
1
rashedcs