c++ - 重载 make_heap？

Question

我有一个 std::vector<t_Mont> v , t_Mont有 {float val, int i and int j}

我想做make_heap(v.begin(), v.end())和 pop_head(v.begin(), v.end())但我在控制台上遇到了很多错误。我想这是因为我传递了一个 t_mont 类型的 vector .

我要make_heap关于变量的值 val的 v .

我必须做什么来编译我？我必须重载 make_heap和 pop_head ？我该怎么做？

谢谢。

我的代码:

std::vector<t_Mont> v;
for (int i = 0; i < nCellsHeight; i++) {
    for (int j = 0; j < nCellsWidth; j++) {
        t_Mont aux;
        aux.i = i;
        aux.j = j;
        aux.val = cellValues[i][j];
        v.push_back(aux);
    }
}

std::make_heap(v.begin(), v.end());
while (v.begin() != v.end()) {
    std::cout << "Best of v = " << v[0].val << std::endl;
    std::pop_heap(v.begin(), v.end());
    v.pop_back();
    std::make_heap(v.begin(), v.end());
}

Answer 1

make_heap默认情况下，相关函数将使用 < 比较值.您需要为您的类型提供该运算符的重载:

bool operator<(t_Mont const & lhs, t_Mont const & rhs) {
    return lhs.val < rhs.val;
}

或在调用函数时提供自定义比较器:

auto comp = [](t_Mont const & lhs, t_Mont const & rhs){return lhs.val < rhs.val;};
std::make_heap(v.begin(), v.end(), comp);

如果您受困于古老的 pre-lambda 编译器，请完整定义一个函数类型:

struct Comp {
    bool operator()(t_Mont const & lhs, t_Mont const & rhs){return lhs.val < rhs.val;}
};

std::make_heap(v.begin(), v.end(), Comp());