使用带有varidic参数的vector :: insert

时间:2015-08-21 13:05:10

标签: c++ templates c++11 vector arguments

我想使用带有varidic参数的模板函数将许多向量组合成一个。 我的三个或更多向量存在问题,低于我当前的代码:

#include <iostream>
#include <vector>
#include <numeric>

template<typename _Ty, typename ..._Args>
std::vector<_Ty> combine(const std::vector<_Ty> &a, const _Args &...args) {
  // Determine size of new vector
  std::vector<std::size_t> sizes = { a.size(), args.size()... };
  std::size_t size = std::accumulate(sizes.begin(), sizes.end(), 0);
  // Create vector with new size
  std::vector<_Ty> result(size);
  // Insert all vectors into this one
  result.insert(a.begin(), a.end(), result.end());
  result.insert(result.end(), args.begin()..., args.end()...);
  return result;
}

int main(int argc, char *argv[], char *envp[]) {
  std::vector<int> a = { 0, 1, 2, 3, 4 };
  std::vector<int> b = { 4, 3, 2, 1, 0 };
  std::vector<int> c = { 1, 1, 1, 1 };

  std::cout << combine(a, b).size() << std::endl;
  std::cout << combine(a, b, c).size() << std::endl; // <-- Does not compile

  std::cin.ignore();

  return 0;
}  

所以确切的问题是组合(a,b,c)不能编译。我知道为什么。因为这一行:

result.insert(result.end(), args.begin()..., args.end()...);

汇编到:

result.insert(result.end(), b.begin(), c.begin(), b.end(), c.end());

但我不知道如何使用varidic参数调用result.insert,因此它将编译为:

result.insert(result.end(), b.begin(), b.end());
result.insert(result.end(), c.begin(), c.end());

一种可能性是:

std::vector<std::vector<_Ty> all = { a, args...};
for (const auto &vec : all) {
    result.insert(vec.begin(), vec.end());
}

但这需要所有载体的第二个副本....任何想法?谢谢!

1 个答案:

答案 0 :(得分:4)

标准技巧是使用扩展器之类的东西:

template<typename _Ty, typename ..._Args>
std::vector<_Ty> combine(std::vector<_Ty> a, const _Args &...args)
                      // ^^^^^^^^^^^^^^^^^^ by-value
{
    using expander = int[];
    expander{0,
        (void(a.insert(a.end(), args.begin(), args.end())), 0)...
    };
    return a;
}

旁注,_Ty_Args是保留名称。