专门针对地图类容器的功能

Question

我想专门为矢量和地图之类的容器创建一个函数模板。对于矢量，我可以像下面那样做，但我不知道我怎样才能有一个专门的函数版本，只用于像容器这样的地图。

#include <iostream>
#include <vector>
#include <map>

using namespace std;

template<typename Iterator>
void print(Iterator begin, Iterator end)
{
    while (begin != end)
    {
        cout << *begin << endl; // compiler error for map like containers
        ++begin;
    }
}

int main()
{
    vector<int> noVec = { 1, 2, 3 };

    print(noVec.begin(), noVec.end());

    map<int, int> nosMap;
    nosMap[0] = 1;
    nosMap[1] = 2;
    nosMap[3] = 3;

    print(nosMap.begin(), nosMap.end());

    return 0;
}

This 问题类似，但它建议在我不想做的向量中使用pair。我知道可以通过 SFINAE 完成专业化，但不知道要检查什么条件。如果我可以使用 C++ 11 type_traits 实现这一点，那就太好了。

Answer 1

map 的value_type 是一对，因此您可以检查迭代器的value_type 是否为std::pair，例如

#include <vector>
#include <map>
#include <iostream>

template <typename> 
struct is_pair : std::false_type 
{ };

template <typename T, typename U>
struct is_pair<std::pair<T, U>> : std::true_type 
{ };



template <typename Iter>
typename std::enable_if<is_pair<typename std::iterator_traits<Iter>::value_type>::value>::type
print(Iter begin, Iter end)
{
  std::cout << "called with map-like" << std::endl;
  for (; begin != end; ++begin)
  {
    std::cout << begin->second;
  }
  std::cout << std::endl;
}

template <typename Iter>
typename std::enable_if<!is_pair<typename std::iterator_traits<Iter>::value_type>::value>::type
print(Iter begin, Iter end)
{
  std::cout << "called with vector-like" << std::endl;
  for (; begin != end; ++begin)
  {
    std::cout << *begin;
  }
  std::cout << std::endl;
}



int main()
{
  std::vector<int> vec { 1, 2, 3 };
  std::map<int, int> map {{0, 0}, {1, 1}, {2, 4}, {3, 9}};

  print(vec.begin(), vec.end());
  print(map.begin(), map.end());
}

打印出来的

called with vector-like
123
called with map-like
0149

Answer 2

你不需要专门化任何东西。您所要做的就是为std::pair 提供重载输出operator<<，如下例所示：

template<typename T1, typename T2>
std::ostream& operator<<(std::ostream &out, std::pair<T1, T2> const &mp) {
  return (out << "(" << mp.first << ", " << mp.second << ")");
}

LIVE DEMO

---

编辑：

然而，如 cmets 中建议的 @Benjamin Lindley ，上述解决方案可能与输出 operator<< 的其他模板重载为 std::pair 冲突。

如果是这种情况，您也可以在自己的命名空间（例如，namespace detail）中编写两个模板函数重载（例如，print_elem），如下例所示：

namespace detail {
  template<typename T1, typename T2>
  std::ostream& print_elem(std::ostream &out, std::pair<T1, T2> const &mp) {
    return (out << "(" << mp.first << ", " << mp.second << ")");
  }

  template<typename T>
  std::ostream& print_elem(std::ostream &out, T const &elem) {
      return (out << elem);
  }
}

并更改您的template print，如下例所示：

template<typename Iterator>
void print(Iterator begin, Iterator end)
{
    while (begin != end) {
        detail::print_elem(cout, *begin) << endl;
        ++begin;
    }
}

LIVE DEMO