TMP：如何概括向量的笛卡尔积？

  • Z时代
  • 2024-01-10
  • 分类：问答

有一个出色的C++解决方案（实际上是2个解决方案：递归和非递归）。为了说明/简单起见，让我们仅关注 。

我的问题是，如何将这些代码与模板通用化，以采用如下所示std::tuple的同类向量：

{{2,5,9},{"foo","bar"}}

并产生一个齐次向量tuple

{{2,"foo"},{2,"bar"},{5,"foo"},{5,"bar"},{9,"foo"},{9,"bar"}}

如果让生活变得更轻松，那么让我们假设输入中的内部向量都是同质的。因此， 这样的输入： {{5,"baz"}{'c',-2}}

将输入从锯齿向量更改为元组

回答：

更简单的递归解决方案。它使用向量作为函数参数，而不是元组。此版本不构建临时元组，而是使用lambda。现在，它不会进行不必要的复制/移动，并且似乎已成功优化。

#include<tuple>
#include<vector>
// cross_imp(f, v...) means "do `f` for each element of cartesian product of v..."
template<typename F>
inline void cross_imp(F f) {
    f();
}
template<typename F, typename H, typename... Ts>
inline void cross_imp(F f, std::vector<H> const& h,
                           std::vector<Ts> const&... t) {
    for(H const& he: h)
        cross_imp([&](Ts const&... ts){
                      f(he, ts...);
                  }, t...);
}
template<typename... Ts>
std::vector<std::tuple<Ts...>> cross(std::vector<Ts> const&... in) {
    std::vector<std::tuple<Ts...>> res;
    cross_imp([&](Ts const&... ts){
                  res.emplace_back(ts...);
              }, in...);
    return res;
}
#include<iostream>
int main() {
    std::vector<int> is = {2,5,9};
    std::vector<char const*> cps = {"foo","bar"};
    std::vector<double> ds = {1.5, 3.14, 2.71};
    auto res = cross(is, cps, ds);
    for(auto& a: res) {
        std::cout << '{' << std::get<0>(a) << ',' <<
                            std::get<1>(a) << ',' <<
                            std::get<2>(a) << "}\n";
    }
}

以上是 TMP：如何概括向量的笛卡尔积？ 的全部内容， 来源链接： utcz.com/qa/405455.html

回到顶部