在 C++ 中迭代 std::map 的键/值

Question

我的问题是对this one 的简单概括。为了便于讨论，我将专注于迭代地图的键。我想要一个通用的辅助函数key_iterator，它接受一个映射迭代器并返回一个映射键迭代器。例如以下代码：

#include "key_iterator.hpp"
#include <algorithm>
#include <iostream>
#include <iterator>
#include <map>

int main(int argc, char** argv)
{
    std::map<std::string, int> m;
    m["One"] = 1;
    m["Two"] = 2;
    std::copy(key_iterator(m.begin()), key_iterator(m.end()), std::ostream_iterator<std::string>(std::cout, " "));
    return 0;
}

应该产生以下输出：

One Two

正如上述问题的解决方案中所建议的，boost::transform_iterator 似乎是实现key_iterator 的合适起点。对于 key_iterator.hpp，我有一个看起来像这样的中途解决方案：

#pragma once

#include <functional>
#include <map>
#include <boost/iterator/transform_iterator.hpp>

template <typename Key, typename Value>
class KeyGetter : public std::unary_function<std::pair<Key,Value>, Key>
{
public:
    const Key& operator()(const std::pair<Key,Value>& p) const {return p.first;}
};

template<typename Key, typename Value>
boost::transform_iterator<KeyGetter<Key,Value>, typename std::map<Key,Value>::iterator>
key_iterator(typename std::map<Key,Value>::iterator itr)
{
    return boost::make_transform_iterator<KeyGetter<Key,Value>, typename std::map<Key,Value>::iterator>(itr, KeyGetter<Key,Value>());
}

但是通过这个实现，Key 和 Value 类型不会被自动推导出来，我需要手动提供它们以使其编译：

std::copy(key_iterator<std::string,int>(m.begin()), key_iterator<std::string,int>(m.end()), std::ostream_iterator<std::string>(std::cout, " "));

对如何按照我想要的方式工作有什么想法吗？

Answer 1

试试这个：

template <typename Iter>
struct KeyGetter : std::unary_function<typename Iter::value_type,
        typename Iter::value_type::first_type>
{
    const typename Iter::value_type::first_type& operator()
            (const typename Iter::value_type& p) const
        { return p.first; }
};

template<typename Iter>
boost::transform_iterator<KeyGetter<Iter>, Iter> key_iterator(Iter itr)
{
    return boost::make_transform_iterator<KeyGetter<Iter>, Iter>
        (itr, KeyGetter<Iter>());
}

这个想法是调用站点的函数应该直接在其参数上进行模板化，以避免必须显式指定模板参数。