STL set::find 重新定义的搜索

Question

我的程序基于一组对，即

typedef std::pair<int,int> innerPair;
typedef std::pair<innerPair,int> setElement;
std::set<setElement> Foo;

innerPair 元素是真正定义 setElement 的元素，但我需要为每个元素附加一个组 ID，因此是后者 setElement 定义。

在程序的其余部分，我需要找到innerPair，而不管他们的组ID，所以我基本上需要一个函数

std::set<setElement>::iterator find(innePair);

无论其组 ID 为何，都会找到 innerPair。就目前而言，我可以简单地循环遍历所有可用的组 ID 并进行多次 find() 调用，但这远没有效率。

有没有一种简洁的方法来定义一个执行某种通配符搜索的find( ... )成员函数，还是我需要用我自己的定义重载它？

Answer 1

如果您有多个元素具有相同的内部对和不同的组 ID，则可以使用 std::multimap<innerPair, int>。

这允许您使用相同的innerPair 存储多个元素。

它还简化了使用lower_bound/upper_bound 或equal_range 的搜索。

Answer 2

我看到了两种可能的设计。一个可能比另一个更适用。

innerPair 可能更适合作为具有 3 个成员（第一、第二和组 ID）的结构或 std::tuple<int, int, int>。组 ID 是 innerPair 对象的一部分吗？如果是这样，那么我建议这是更好的设计。

如果不是（老实说，我认为在您的情况下不是），您应该使用 std::map<innerPair,int> 创建从 innerPair 对象到组 ID 的映射。然后你可以很容易地找到一个元素：

std::map<innerPair,int> mapping;
// Fill your map
innerPair key = {1, 2};
auto found_iter = mapping.find(key);

您还可以通过以下方式获取特定 innerPair 的组 ID：

int group_id = mapping[key];

您不需要提供自定义比较器，因为 operator< 已为 std::pair 定义。

Answer 3

如果您想按部分对象进行搜索，最好不要使用std::set<...>，而是使用std::map<...>：

std::map<std::pair<int, int>, int>

无论如何，这几乎具有您为std::set<...> 定位的值类型（不同之处在于first 成员被声明为const）。

如果您真的坚持使用std::set<...>，则需要创建一个忽略最后一个second 成员的比较器类型。我不确定std::set<...> 是否支持混合类型比较（我认为它不支持）。

Answer 4

您可以使用multiset，例如带有自定义比较函数或仿函数：

struct innerPairCompare
{
    bool operator () (const setElement &a, const setElement &b)
    {
        const innerPair &a_ = a.first;
        const innerPair &b_ = b.first;
        return (a_.first > b_.first || a_.first = b_.first && a_.second > b_.second);
    }

};

然后将其用于您的multiset:

std::multiset<setElement,innerPairCompare> Foo;

它将所有具有相同innerPair 的setElements 存储在同一个列表中。

或者，如果您需要具有给定GroupID 的所有innerPair，请使用比较groupID 的函数。

最后，如果您真的不需要将GroupID 和innerPair 放在一起，您可以使用map（或multimap），使用GroupID 或innerPair 作为Key。

Answer 5

如果您想继续使用std::set，那么您可以使用带有自定义谓词的std::find_if，看看这个answer。

基本上你会定义一个函数

bool pairCorresponds(std::pair<int,int> element)

这将为您完成工作。