有没有STL函数可以得到两个C++向量的笛卡尔乘积

Is there some STL function to get cartesian product of two C++ vectors?

本文关键字:C++ 两个 向量 笛卡尔 函数 STL 有没有      更新时间:2023-10-16

假设

b =  ["good ", "bad "]
a  = ["apple","mango"]
then output = ["good apple","good mango","bad apple","bad mango"]

我知道这可以用嵌套的for循环来完成,但有没有一些优雅的一行代码可以用C++STL来完成呢?

这里有一行(复制自Jonathan Mee在这里发布的答案):

for(size_t i = 0, s = a.size(); i < output.size(); ++i) output[i] = b[i/s] + ' ' + a[i%s];

这里是完整的示例。

给定vector<string> avector<string> b,您可以使用for_each:

vector<string> output(size(a) * size(b));
for_each(begin(output), end(output), [&, it = 0U](auto& i) mutable {
i = a[it / size(b)] + ' ' + b[it % size(b)];
++it;
});

实时示例

编辑:

我们已经初始化了output,有足够的空间容纳ab的每一个组合。然后我们将遍历output的每个元素并分配它

我们将使用a的1st元素作为output的第一个size(b)元素,使用a的2nd元件作为第二个size(b)元素,依此类推。因此,我们将通过使用it / size(b)进行索引来实现这一点。我们希望通过b元素的迭代将其结合起来。

it将移动到output的每个元素的下一个索引,但索引需要换行,否则当it == size(b)时它将越界,为此我们使用it % size(b)

第2版:

在这个问题中,通过基准测试,我发现了一个现象,即模和除法是迭代的昂贵操作。我在这里做了同样的测试。为了隔离算法,我只是在vector<int>而不是vector<string>上进行笛卡尔求和。

首先,我们可以看到这两种算法导致不同的组装。我上面写的算法需要585行汇编。根据我对MSalter代码的解释,需要588行

vector<string> output(size(testValues1) * size(testValues2));
auto i = begin(output);
std::for_each(cbegin(a), cend(a), [&](const auto& A) { std::for_each(cbegin(b), cend(b), [&](const auto& B) { *i++ = A + ' ' + B; }); });

我在这里放置了一个相当可靠的基准测试:http://ideone.com/1YpzIO在测试中,我只设置了100次测试,但MSchanges的算法总是获胜。在本地使用Visual Studio 2015发布的10000000次测试中,MSchanges算法的完成时间大约是我的三分之二。

很明显,模不是一个很好的索引方法:(

没有直接的解决方案;我检查了整个<algorithm>。没有一个函数产生长度为M*N的输出。

可以调用第一个范围上的std::for_each,使用在第二个范围上调用std::for_each的lambda(!)

std::vector<std::string> a, b;
std::for_each(a.begin(), a.end(), 
[&](std::string A) { std::for_each(b.begin(), b.end(),
[A](std::string B) { std::cout << A << '/' << B << 'n';  }
);});

但这只是STL中的一个嵌套循环。