如何通过 STL 算法获得两个数组（向量）的组合？

【问题标题】：How to get combination of two array ( vector ) by STL algorithm?如何通过 STL 算法获得两个数组（向量）的组合？
【发布时间】：2009-10-10 00:43:25
【问题描述】：

我有 v1 和 v2 ，我应该如何获得一个新的 v 如下所示？

v1 = {1,2}
v2 = {3,4,5}

v = {f(1,3) , f(1,4) , f(1,5) f(2,3) ,f(2,4) ,f(2,5)}

我知道我可以使用两个循环来做到这一点，但是如果有更惯用的方式，比如使用 STL 算法？

//using two loops
for iter1 of v1 
     for iter2 of v2
           v.push_back(f(v1,v2))

编辑：

v1 和 v2 不必具有相同的大小。

【问题讨论】：

有什么好的非递归算法来计算笛卡尔积？：stackoverflow.com/questions/215908/…
这看起来几乎完全符合您的要求：http://stackoverflow.com/questions/979436

标签： c++ stl vector combinations

【解决方案1】：

没有合适的 STL 算法，但这种组合可以通过自定义函数和 std::generate:

#include <vector>
#include <algorithm>
#include <iostream>
#include <iterator>

typedef int T;

struct Fctor
{
    typedef std::vector<T>::iterator Iterator;
    Iterator it1, it2, begin, end;

    Fctor(Iterator begin1, Iterator end1, Iterator begin2)
    {
        begin = begin1;
        end = end1;
        it1 = begin1;
        it2 = begin2;
    }

    T operator()()
    {
        // T result = f(*it1, *it2);
        T result = (*it1) * (*it2);

        if(++it1 != end) return result;
        it1 = begin;
        ++ it2;
        return result;
    }
};


int main()
{
    std::vector<T> v1; v1.push_back(1); v1.push_back(2);
    std::vector<T> v2; v2.push_back(3); v2.push_back(4); v2.push_back(5);
    std::vector<T> result(v1.size() * v2.size());

    Fctor fctor(v2.begin(), v2.end(), v1.begin());
    generate(result.begin(), result.end(), fctor);

    std::copy(result.begin(), result.end(), 
              std::ostream_iterator<T>(std::cout, " "));
    std::cout << std::endl;
    // => 3 4 5 6 8 10
}

【讨论】：