c++ - 在 std::set 中按键查找

Question

我正在努力std::set支持自定义比较操作。我的目标是能够仅使用 key 参数调用 find ，而无需先创建 set 类型的对象。像这样的东西：

#include <set>
#include <cstdint>
#include <vector>

class TestItem {
  public:
    std::uint64_t id;          // key
    std::vector<double> areas; // some other data
};

struct TestItemCompare {
    bool operator()(const std::uint64_t& lhs, const std::uint64_t& rhs) {
        return lhs < rhs;
    }
    bool operator()(const std::uint64_t& lhs, const TestItem& rhs) {
        return lhs < rhs.id;
    }
    bool operator()(const TestItem& lhs, const std::uint64_t& rhs) {
        return lhs.id < rhs;
    }
    bool operator()(const TestItem& lhs, const TestItem& rhs) {
        return lhs.id < rhs.id;
    }
};

int main() {
    std::set<TestItem, TestItemCompare> store;
    std::uint64_t id = 0;
    TestItem t;
    t.id = 0;

    auto it1 = store.find(t);    // compiles
    auto it2 = store.find(id);   // fails to compile

    return 0;
}

我以为这是可能的...

标签： c++stlsetc++17

Answer 1

有可能的。但是您的比较器必须是透明的。而且functor中的成员函数必须是const

struct TestItemCompare {
    using is_transparent = int; // actually any alias will do, void, custom type
    bool operator()(const std::uint64_t& lhs, const std::uint64_t& rhs) const {
        return lhs < rhs;
    }
    bool operator()(const std::uint64_t& lhs, const TestItem& rhs) const {
        return lhs < rhs.id;
    }
    bool operator()(const TestItem& lhs, const std::uint64_t& rhs) const {
        return lhs.id < rhs;
    }
    bool operator()(const TestItem& lhs, const TestItem& rhs) const {
        return lhs.id < rhs.id;
    }
};