如何使std::pop_heap接受适当的函数指针作为Compare

How to make appropriate function pointer accepted by std::pop_heap as Compare?

本文关键字:函数 指针 Compare std 何使 pop heap      更新时间:2023-10-16

我有一个Node类,函数为

static bool HasGreaterF(const Node& a, const Node& b);
static bool HasGreaterF(const std::shared_ptr<Node>& a, const std::shared_ptr<Node>& b);

然后我将一个向量作为堆,并希望使用第二个函数作为Compare。这不起作用,因为函数指针无法与一个函数匹配。如果我删除第一个函数,它就会工作。如何使它同时使用这两种功能?

std::vector<std::shared_ptr<Node>> openlist_min_f;
std::pop_heap(openlist_min_f.begin(), openlist_min_f.end(), &Node::HasGreaterF);

使用static_cast指定它:

std::pop_heap(openlist_min_f.begin(), openlist_min_f.end(), 
              static_cast<bool (*)(const std::shared_ptr<Node>&, const std::shared_ptr<Node>&)>
              (&Node::HasGreaterF));

或者像@David建议的那样使用lambda包装器,因为在这种情况下函数不应该是模糊的。

std::pop_heap(openlist_min_f.begin(), openlist_min_f.end(), 
              [](const auto& l, const auto& r){ return Node::HasGreaterF(l, r); });

不能按原样使用它的原因是pop_heap必须推导出您传入的比较器的类型,而重载函数没有一个类型。在某些上下文中,除了简单地调用重载函数外,还可以使用重载函数的名称,但作为参数传递给函数模板并不是其中之一。在这种情况下,您必须明确声明您所指的HasGreaterF

您可以使用static_cast(如前所述),也可以将其封装在lambda:中

std::pop_heap(openlist_min_f.begin(), openlist_min_f.end(),
    [](const auto& lhs, const auto& rhs) { return Node::HasGreaterF(lhs, rhs); });

lambda的优点在于,根据我的经验,它更有可能被内联(尽管两者都非常冗长)。您总是可以在C++14:中宏化lambda包装

#define WRAP_FUN(f) [](auto&&... args) -> decltype(auto) { return f(std::forward<decltype(args)>(args)...); }

从而变成:

std::pop_heap(openlist_min_f.begin(), openlist_min_f.end(), WRAP_FUN(Node::HasGreaterF));