C ++ STL make_heap и pop_heap не работают

Question

Мне нужно использовать кучу, поэтому я искал STL, но она не работает, я написал некоторый код, чтобы объяснить, что я имею в виду:

#include <stdio.h>
#include <stdlib.h>
#include <vector>
#include <algorithm>

struct data
{
    int indice;
    int tamanho;
};


bool comparator2(const data* a, const data* b)
{
    return (a->tamanho < b->tamanho);
}

int main()
{

        std::vector<data*> mesas;

        data x1, x2, x3, x4, x5;

        x1.indice = 1;
        x1.tamanho = 3;

        x2.indice = 2;
        x2.tamanho = 5;

        x3.indice = 3;
        x3.tamanho = 2;

        x4.indice = 4;
        x4.tamanho = 6;

        x5.indice = 5;
        x5.tamanho = 4;

        mesas.push_back(&x1);

        mesas.push_back(&x2);

        mesas.push_back(&x3);

        mesas.push_back(&x4);

        mesas.push_back(&x5);


        make_heap(mesas.begin(), mesas.end(), comparator2);

        for(int i = 0 ; i < 5 ; i++)
        {
            data* mesa = mesas.front();
            pop_heap(mesas.begin(),mesas.end());
            mesas.pop_back();

            printf("%d, %d\n", mesa->indice, mesa->tamanho);
        }

    return 0;
};

и вот что я получаю:

4, 6
2, 5
1, 3
3, 2
5, 4

Так что он не работает как куча, так как максимальный элемент вектора не возвращается правильно.

Или я что-то не так делаю?

Answer 1

Вам нужно передать comparator2 в std::pop_heap, поскольку именно так вы и создали кучу. В противном случае он будет использовать по умолчанию оператор меньше чем для указателей, который просто сравнивает значения указателя.

Answer 2

MSN ответ правильный. Тем не менее, одно из двух правил стиля может предотвратить эту ошибку:

• Объявите компаратор для работы с ссылками, а не с объектами, как operator<. Используйте vector объектов, а не указатели.

  bool comparator2(const data& a, const data& b)
  {
      return (a.tamanho < b.tamanho);
  }
  

  Вам может действительно понадобиться вектор указателей, в этом случае это не применимо.

• Используйте std::priority_queue (из <queue>), который связывает вместе pop_heap и pop_back для вас, помня ваш компаратор. Для этого требуется компаратор функтора:

  struct comparator2 { bool operator()(const data& a, const data& b)
  {
      return (a.tamanho < b.tamanho);
  } };
   
  std::priority_queue<data, vector<data>, comparator2> mesas;
   // or std::priority_queue<data, vector<data>, comparator2>
  mesas.push(x1);
  

• Самый элегантный способ - сделать это сравнение по умолчанию для data:

  struct data
  {
      int indice;
      int tamanho;
       
      friend bool operator<(const data& a, const data& b)
      {
          return (a.tamanho < b.tamanho);
      }
  };
  std::priority_queue<data> mesas;
  mesas.push(x1);
  

priority_queue также может взять предварительно заполненный несортированный контейнер, который будет скопирован.

Answer 3

У меня была похожая проблема, и я смог ее решить примерно так:

struct comparator2 { bool operator()(data const * const a, data const * const b)
{
    return (a->tamanho < b->tamanho);
 } };

std::priority_queue<data*, std::vector<data*>, comparator2> mesas;

Answer 4

А как насчет std :: set

#include <stdio.h>
#include <stdlib.h>
#include <vector>
#include <algorithm>
#include <set>

struct data
{
    // Always put constructors on.
    // When the constructor is finished the object is ready to be used.
    data(int i,int t)
        :indice(i)
        ,tamanho(t)
    {}

    int indice;
    int tamanho;

    // Add the comparator to the class.
    // Then you know where to look for it.
    bool operator<(data const& b) const
    {
        return (tamanho < b.tamanho);
    }
};



int main()
{

        std::set<data> mesas;

        // Dont declare all your variables on the same line.
        // One per line otherwise it is hard to read.
        data x1(1,3);
        data x2(2,5);
        data x3(3,2);
        data x4(4,6);
        data x5(5,4);

        mesas.insert(x1);
        mesas.insert(x2);
        mesas.insert(x3);
        mesas.insert(x4);
        mesas.insert(x5);
        // You don't actually need the variables.
        // You could have done it in place.
        mesas.insert(data(6,100));

        // Use iterator to loop over containers.
        for(std::set<data>::iterator loop = mesas.begin(); loop != mesas.end(); ++loop)
        {
            printf("%d, %d\n", loop->indice, loop->tamanho);
        }

    return 0;
};