维护std :: priority_queue的堆属性

时间:2013-10-09 18:45:17

标签: c++ data-structures stl priority-queue

我希望根据边缘权重创建一个最大堆。但是,在打印堆时,很明显它没有维护其二进制堆属性。我该如何处理?

#include <iostream>
#include <cstring>
#include <sstream>
#include <fstream>
#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <vector>
#include <set>
#include <map>
#include <bitset>
#include <climits>
#include <ctime>
#include <algorithm>
#include <functional>
#include <stack>
#include <queue>
#include <list>
#include <deque>
#include <sys/time.h>
#include <iomanip>
#include <cstdarg>
#include <utility> //std::pair
#include <cassert>
using namespace std;


typedef struct{
        int u, v, weight;
    }edge;


class comparator{
public:
    bool operator()(const edge &e1, const edge &e2)
        {
            if(e1.weight<e2.weight)
                return true;
        }
};


void print_queue(priority_queue<edge, vector<edge> ,comparator> q)
{
    while(!q.empty())
        cout<<q.top().weight<<" ", q.pop();
    cout<<endl<<endl;
}

int main()
{

    priority_queue<edge, vector<edge> , comparator> q;
    edge e1={1,2,10};
    edge e2={1,3,23};
    edge e3={1,4,4};
    edge e4={1,5,99};
    edge e5={1,6,43};
    edge e6={1,7,29};

    q.push(e1);
    q.push(e2);
    q.push(e3);
    print_queue(q);
    q.push(e4);
    q.push(e5);
    q.push(e6);
    print_queue(q);
}

2 个答案:

答案 0 :(得分:3)

该错误发生在您的比较器功能中。在e1.weight&gt; = e2.weight:

的情况下,我添加了“return false”
    bool operator()(const edge &e1, const edge &e2)
    {
        if(e1.weight<e2.weight)
            return true;
        return false;
    }

我觉得它现在按照你期望的方式运作:

在:

  

[jsaxton @ jsaxton-dev~] $ ./a.out   4 23 10

     

29 23 4 10 43 99

后:

  

[jsaxton @ jsaxton-dev~] $ ./a.out   23 10 4

     

99 43 29 23 10 4

答案 1 :(得分:1)

if(e1.weight<e2.weight)
       return true;

好 其他部分呢?

工作正常
else return false;