web-dev-qa-db-ja.com

ユーザー定義タイプの優先キュー

私は以下の構造体を持っています

struct node{
   float val;
   int count;

}

この構造体のオブジェクトがいくつかあります。ここで、これらのオブジェクトをSTLの優先キューに挿入して、優先キューがアイテムをカウント順に並べるようにします。そうする方法について何かアイデアはありますか?最小ヒープが好ましい。構造体ではなく、プリミティブデータ型に対して上記を行う方法を知っています

11
Programmer

<演算子をオーバーロードします。

bool operator<(const node& a, const node& b) {
  return a.count > b.count;
}

優先キューに追加の引数を渡さずに最小ヒープを達成するために、比較を逆にしました。今、あなたはそれをこのように使います:

priority_queue<node> pq;
...

編集:ほぼ正確に重複しているように見えるこの投稿を見てください: カスタムクラスのSTL優先度キュー

16
#include <iostream>
#include <queue>
#include <vector>
using namespace std;
class Boxer{
    public:
        string name;
        int strength;
};
struct Comp{
    bool operator()(const Boxer& a, const Boxer& b){
        return a.strength<b.strength;
    }
};
int main(){
    Boxer boxer[3];
    boxer[0].name="uday", boxer[0].strength=23;
    boxer[1].name="manoj", boxer[1].strength=33;
    boxer[2].name="rajiv", boxer[2].strength=53;

    priority_queue< Boxer, vector<Boxer>, Comp> pq;
    pq.Push(boxer[0]);
    pq.Push(boxer[1]);
    pq.Push(boxer[2]);
    Boxer b = pq.top();
    cout<<b.name;
    //result is Rajiv

    return 0;
}
10
Uday Hiwarale
  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;
        }
    
  5. 演算子のオーバーロードでも同じ結果が得られます。

    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;
    }
4
Taohidul Islam

その構造体にoperator<を指定する必要があります。何かのようなもの:

bool operator<(node const& x, node const& y) {
    return x.count < y.count;
}

これで、標準ライブラリの優先キューを使用できます。

C++ 11以降、次のように記述できます。

auto comparer = [](const auto& a, const auto& b) {
    return a.priority < b.priority;
};
std::priority_queue<Item, std::vector<Item>, decltype(comparer)> queue(comparer);
0
yukinarit

ユーザー定義のコンパレータクラスを定義できます。

コードスニペット :

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

struct man
{
  string name;
  int priority; 
};

class comparator
{
 public:
   bool operator()(const man& a, const man& b)
   {
        return a.priority<b.priority;
   }
};

int main()
{
   man arr[5];
   priority_queue<man, vector<man>, comparator> pq;

   for(int i=0; i<3; i++)
   {
     cin>>arr[i].name>>arr[i].priority;
     pq.Push(arr[i]);
   }

   while (!pq.empty())
   {
     cout<<pq.top().name<<" "<<pq.top().priority;
     pq.pop();
     cout<<endl;
   }
   return 0;
}
0
rashedcs