创建包含唯一元素的优先级队列

时间:2015-02-22 12:09:54

标签: c++ vector

我想创建一个包含唯一元素的优先级队列。我可以发现包含唯一向量元素的队列可以按如下方式创建:

template <typename T>
class queue1 {
private:
    std::queue<T> m_queue;
    std::set<T> m_set;
public:
    bool push(const T& t) {
        if (m_set.insert(t).second) {
            m_queue.push(t);
            return true;
        }
        return false;
    }
};

queue1<vector<unsigned> > queueUnique;

但我的要求是除了包含唯一向量元素的队列之外,它应该是一个优先级队列,因为队列中的每个向量都有一个与之关联的分数。我尝试使用以下方法创建此队列:

priority_queue<queueUnique, vector<queueUnique>, ComparisonFunction> pq;

然而,这似乎是不正确的,因为它提供了一个队列队列。

我不知道如何将得分与每个向量相关联,这样我得到的结果队列就是具有唯一向量元素的优先级队列。例如,如果我的队列是:

struct myDS{
  vector<unsigned> vec;
  double score;
};
queue<myDS> myqueue;
vector<unsigned> dummyVec1;
dummyVec1.push_back(5);
myDS obj;
obj.vec=dummyVec1;
obj.score=0.9;
vector<unsigned> dummyVec2;
dummyVec2.push_back(5);
myDS obj2;
obj2.vec=dummyVec2;
obj2.score=0.9; //with duplicate values score is always same so can be eliminated

然后,我如何创建一个在obj.score上排序但不包含重复元素的优先级队列。例如,上面的myqueue应包含&#34; 5&#34;只有一次。

2 个答案:

答案 0 :(得分:2)

鉴于与每个std::vector相关联的分数是唯一的,只需向struct myDS添加比较器功能即可,以便std::priority_queue可以确定排序:

这样的事情:

struct myDS
{
    std::vector<unsigned> vec;
    double score;

    // Comparison function for ordering
    // based on score 
    bool operator<(const myDS& rhs) const
    {
        return score < rhs.score;
    }
};

template<typename T>
class queue1
{
private:

    std::priority_queue<T> m_queue;
    std::set<T> m_set;
public:
    bool push(const T& t)
    {
        if(m_set.insert(t).second)
        {
            m_queue.push(t);
            return true;
        }
        return false;
    }
};

queue1<myDS> queueUnique;

答案 1 :(得分:0)

正如Galik评论的那样 - 您正在寻找的是一个有序的集合:

#include <set>

struct ComparableThingy{
  ComparableThingy(std::string const& n, unsigned v):name(n),value(v){}
  std::string name;
  unsigned value;
  bool operator <(ComparableThingy const& other)const{return value<other.value;}
};

inline std::ostream& operator<<(std::ostream& out, std::set<ComparableThingy> const& s){
   for(auto const& v:s) out << v.name << "," << v.value << " ";
   return out;
}

int main()
{
 std::set<ComparableThingy> s;

 s.emplace("a",3);
 s.emplace("b",2);
 s.emplace("c",3);
 s.emplace("d",1);

 std::cout << s << "\n";

 // Equivalent to "pop"
 std::cout << "pop...\n";
 s.erase(s.begin());

 std::cout << s << "\n";
}

输出:

d,1 b,2 a,3   
pop...
b,2 a,3