如何获得 std::map 的 std::set 键

Question

今天早上我正在写一个算法，我遇到了一个奇怪的情况。 我有两个std::map 。 我想对每个键的集合执行集合交集（以查找两个映射共有的键）。 在未来的某个时候，我想我很可能也想在这里执行集合减法。 幸运的是，STL 包含这两种操作的功能。 问题是，我似乎无法从std::map中得到一个std::set键。 有没有办法做到这一点？ 我正在寻找这样简单的东西，就像在 Java 中一样：

std::set<Foo> keys = myMap.getKeySet();

我的理解是，我不能直接在映射中的迭代器上使用std::set_intersection()函数，因为映射公开std::pair对象而不仅仅是键。 另外，我认为地图不能保证秩序。 我也有兴趣在一对std::multimap上执行相同的操作，如果这有什么不同的话。

编辑：我最初忘了提到，由于我被迫使用的编译器的年龄（MSVC++ 6），大多数在 boost 中可用的漂亮模板技巧都不能使用。

Answer 1

你基本上想要的是一个副本，因为std :: map不会将键保存在std :: set中。 std :: copy假定值类型是兼容的，这不是这里的情况。 std :: map :: value_type是一个std :: pair。 您只想复制该对的第一部分，这意味着您需要一个std :: transform。 现在，由于您将在集合上使用insert_iterator，因此顺序无关紧要。 即使地图已经排序，std :: set也会对插入进行排序。

[编辑]代码可能更容易。 我的头脑，没有编译。

std::transform(MyMap.begin(), MyMap.end(),
    std::inserter(MySet, MySet.end()),
    boost::bind(&std::pair<Key,Value>::first, _1));

如果你有SGI的select1st，你不需要boost :: bind。

[edit]更新了C ++ 14

std::transform(MyMap.begin(), MyMap.end(),
    std::inserter(MySet, MySet.end()),
    [](auto pair){ return pair.first; });

Answer 2

地图确实保证秩序; 这就是为什么它被称为有序关联容器 。 您可以将set_intersection与自定义比较器功能一起使用，这是此处列出的第二个变体。

所以，像

bool your_less(const your_map::value_type &v1, const your_map::value_type &v2)
{ return v1.first < v2.first; }

set_intersection(m1.begin(), m1.end(), m2.begin(), m2.end(), your_output_it, your_less);

应该做的伎俩。 （也可以使用boost :: lambda和bind来避免编写临时函数。）

默认运算符<over pairs比较两个组件。 由于您只需要在对的第一部分（映射键）上进行等效，因此您需要定义自己的比较运算符来提供这种关系（这是上面的函数所做的）。

Answer 3

您可以使用通用的boost :: transform_iterator返回仅返回键（而不是值）的迭代器。 请参阅如何从std :: map中检索所有键（或值）并将它们放入向量中？

Answer 4

在实践中，

yourmap::const_iterator mi;
set<key_type> k;
for (mi = yourmap.begin(); mi != yourmap.end(); ++mi)
  k.insert(mi->first);
return k; 

Answer 5

最好的非SGI，非升级STL算法友好的解决方案是扩展map :: iterator，如下所示：

template<typename map_type>
class key_iterator : public map_type::iterator
{
public:
    typedef typename map_type::iterator map_iterator;
    typedef typename map_iterator::value_type::first_type key_type;

    key_iterator(const map_iterator& other) : map_type::iterator(other) {} ;

    key_type& operator *()
    {
        return map_type::iterator::operator*().first;
    }
};

// helpers to create iterators easier:
template<typename map_type>
key_iterator<map_type> key_begin(map_type& m)
{
    return key_iterator<map_type>(m.begin());
}
template<typename map_type>
key_iterator<map_type> key_end(map_type& m)
{
    return key_iterator<map_type>(m.end());
}

然后像这样使用它们：

        map<string,int> test;
        test["one"] = 1;
        test["two"] = 2;

        set<string> keys;

//      // method one
//      key_iterator<map<string,int> > kb(test.begin());
//      key_iterator<map<string,int> > ke(test.end());
//      keys.insert(kb, ke);

//      // method two
//      keys.insert(
//           key_iterator<map<string,int> >(test.begin()),
//           key_iterator<map<string,int> >(test.end()));

        // method three (with helpers)
        keys.insert(key_begin(test), key_end(test));

Answer 6

您可以迭代并将每个键添加到集合中。 集合和映射都是有序的，无序变量不是。

Answer 7

我在这里找到了你的问题的良好链接

并为您的问题提供一些代码：

    #include <iostream>
    #include <map>
    #include <set>
    #include <iterator>

    typedef std::map<std::string, int> MyMap;

    // also known as select1st in SGI STL implementation
    template<typename T_PAIR>
    struct GetKey: public std::unary_function<T_PAIR, typename T_PAIR::first_type>
    {
        const typename T_PAIR::first_type& operator()(const T_PAIR& item) const
        {
            return item.first;
        }
    };

    int main(int argc, char** argv)
    {
        MyMap m1,m2;

        m1["a"] = 1;
        m1["b"] = 2;
        m2["c"] = 3;
        m2["b"] = 3;

        std::set<std::string> s;
        std::transform(m1.begin(), m1.end(), std::inserter(s, s.begin()), GetKey<MyMap::value_type>());
        std::transform(m2.begin(), m2.end(), std::inserter(s, s.begin()), GetKey<MyMap::value_type>());
        std::copy(s.begin(), s.end(), std::ostream_iterator<std::string>(std::cout, " "));
        std::cout << std::endl;
        return 0;
    }

Answer 8

您也许可以为地图创建一个迭代器，它只使用boost :: adapters :: map_key生成键，请参阅boost :: adapters :: map_key文档中的示例。 这似乎是在Boost 1.43中引入的，并且应该与C ++ 2003兼容，但我不知道VC ++ 6是否具体，这是来自C ++ 98时代。

Answer 9

从zvrba的答案和dianot的评论中建立起来：

只需使接收集合成为对的矢量而不是地图，而dianot指出的问题就结束了。 所以，使用zvrba示例：

std::vector<std::pair<keytype, valtype>> v;

set_intersection(m1.begin(), m1.end(), m2.begin(), m2.end(),
std::back_inserter(v), []( std::pair<keytype, valtype> const & a,
std::pair<keytype, valtype> const & b){return a.first < b.first;});

因此，如果不构建中间副本或集合，我们可以有效地获得两个映射的交集。 这个结构用gcc5.3编译。

Answer 10

我有同样的问题。 我通过使用函数模板解决了它

template <class T, class U>
set<T> getSetFromKeys(map<T,U> mapWithKeys){

  set<T> outputSet;

  for (pair<const T, U> keyValuePair : mapWithKeys) {
    outputSet.insert(keyValuePair.first);
  }

  return outputSet; 
}

通过这种方式，我有一种从应用程序中的映射中提取键的通用方法。

Answer 11

正如 MSalters 所说，您可能只想创建一组密钥。 这对于小地图来说可能已经足够了。

一个快速的方法是：

std::set<Foo> keys;
for (auto &iter : myMap)
{
   keys.insert(iter.first);
}