通过匹配键的子元素来查找地图中的任何元素

Question

我有一个std::map，它的键是另一个类。像下面的例子：

class KeyClass
{
public:
  int a;
  int b;
};

main(){
    //some code
    std::map<KeyClass, SomeOtherClass> mapVariable;
    KeyClass k1(); k1.a = 1; k1.b = 1;
    KeyClass k2(); k2.a = 2; k2.b = 2;
    mapVariable[k1] = SomeOtherClass();
    mapVariable[k2] = SomeOtherClass();

    //I am trying something like see if any element is there whose key.a = 1 (or may be key.a = 6 --> will fail here)?
}

我尝试遍历mapVariable 并检查iter->first.a == 1，但无论如何我可以使用map::count() 函数来获得它吗？

我在 CLI C++ 代码中使用它，所以我似乎无法使用 lambda 函数。

Answer 1

您可以使用std::find_if() 使用自定义谓词搜索地图，例如：

#include <algorithm>

using myMapType = std::map<KeyClass, SomeOtherClass>;
myMapType mapVariable;
...
auto iter = std::find_if(
    mapVariable.begin(), mapVariable.end(),
    [](const myMapType::value_type &entry){ return entry.first.a == 1; }
);
if (iter != mapVariable.end()) {
    ...
}

如果不能使用 lambda，您可以使用仿函数：

#include <algorithm>

typedef std::map<KeyClass, SomeOtherClass> myMapType;

struct findKey
{
    bool operator()(const myMapType::value_type &entry) const { return entry.first.a == 1; }
};

myMapType mapVariable;
...
myMapType::iterator iter = std::find_if(mapVariable.begin(), mapVariable.end(), findKey());
if (iter != mapVariable.end()) {
    ...
}

如果不想使用std::find_if()，可以使用手动循环：

myMapType::iterator iter = mapVariable.begin();
while (iter != mapVariable.end()) {
    if (iter->first.a == 1) break;
    ++iter;
}
if (iter != mapVariable.end()) {
    ...
}

Answer 2

您可以将std::find_if 与 lambda 一起使用：

auto it = std::find_if(mapVariable.begin(), mapVariable.end(), [](const auto& pair) {
    return pair.first.a == 1; // first is the key, second is the value
});

或者不使用 lambda：

struct Functor {
    bool operator()(const decltype(mapVariable)::value_type& pair) const {
        return pair.first.a == 1;
    }
};

auto it = std::find_if(mapVariable.begin(), mapVariable.end(), Functor{});

然后检查是否匹配：

if(it != mapVariable.end()) {
    // match found
}

为了能够使用size_type std::map::count( const K& x ) const 重载，地图中使用的比较函数需要是透明的。

例子：

struct Compare {
    using is_transparent = int;

    bool operator()(int x, const KeyClass& b) const {
        return x < b.a;
    }
    bool operator()(const KeyClass& a, int x) const {
        return a.a < x;
    }
    bool operator()(const KeyClass& a, const KeyClass& b) const {
        return a.a < b.a;
    }
};

//...
    std::map<KeyClass, SomeOtherClass, Compare> mapVariable;
    //...
    if(mapVariable.count(1)) {
        // match found
    }