c++でキューに2つの値を持った変数を入れたい

Question

実現したいこと

ID1, ID2, 値
のデータを値でソートした状態でstd::queueを利用したいです。
具体的には、値が小さいものから順番にID1とID2を取り出したいです
また、queueから順番に取る最中に、新しいデータを追加する予定です。

わからない部分

std::queue単体の、1つだけの変数を使う使い方はネット上にたくさんサンプルコードがあったのでわかりましたが、上のように3つの値を格納したい場合はどうすればよいかがわかりません。

よろしくおねがいします。

Answer 1

こんなの?

#include <iostream>
#include <queue>

struct record {
  int ID1;
  int ID2;
  int value;
};

inline bool operator<(const record& x, const record& y) {
  return x.value > y.value;
}

std::ostream& operator<<(std::ostream& stream, const record& r) {
  return stream << r.ID1 << '/' << r.ID2 << '/' << r.value; 
}

int main() {
  std::priority_queue<record> q;
  for ( int i=0; i < 30; ++i ) {
    record r;
    r.ID1 = i % 7;
    r.ID2 = i % 5;
    r.value = i % 3;
    q.push(r);
  }
  
  while ( !q.empty() ) {
    std::cout << q.top() << std::endl;
    q.pop();
  }
}

[追記] 比較関数を与えた版

#include <iostream>
#include <queue>

struct record {
  int ID1;
  int ID2;
  int value;
};

struct greater_record {
  bool operator()(const record& x, const record& y) const {
    return x.value > y.value;
  }
};

std::ostream& operator<<(std::ostream& stream, const record& r) {
  return stream << r.ID1 << '/' << r.ID2 << '/' << r.value; 
}

int main() {
  std::priority_queue<record, std::vector<record>, greater_record> q;
  for ( int i=0; i < 30; ++i ) {
    record r;
    r.ID1 = i % 7;
    r.ID2 = i % 5;
    r.value = i % 3;
    q.push(r);
  }
  
  while ( !q.empty() ) {
    std::cout << q.top() << std::endl;
    q.pop();
  }
}

Comments

[fdaskjlfda]

ありがとうございます！

[episteme]

注意: 同じvalueの複数のrecordがpushされたとき、取り出される順は不定となります。

[fdaskjlfda]

了解いたしました！本当にどうもありがとうございます！

[yohhoy]

質問にある要件はみたすものの、recordクラスの operator<(より小さい) 実装が > (より大きい) の評価結果を返すのは、潜在的なトラブルのもとになりそうですね...

std::priority_queue<T,Container,Compare> テンプレートの Compare にカスタム関数オブジェクトを指定するほうが、全体のコード設計としてはスマートな気がします。
https://cpprefjp.github.io/reference/queue/priority_queue.html

[episteme]

priority_queueで小さい順に取り出すとなると < の意味を逆転させんならんので。
ただそれは直観に背くので、Compareいぢくる方が素直ですわなー

# 追記しときました (2012/02/25)

[fdaskjlfda]

ありがとうございます！