如何从不同的组成员获得笛卡尔积?

时间:2015-08-17 11:35:09

标签: c++ algorithm vector cartesian-product

正如标题所述,我在C ++中遇到了一些问题。

如果我有std::vector<std::vector<int> >tmpvec

vector < vector <int> > tmpvec = {
    {1,2,3},
    {4,5},
    {6,7,8,9},
    {10,11}
};

如何生成矢量矢量

的所有可能组合
1,4,6,10

1,4,6,11

1,4,7,10

1,4,7,11

......

最多有48种不同的组合。

1 个答案:

答案 0 :(得分:0)

你可以做

bool increase(const std::vector<std::vector<int>>& v, std::vector<std::size_t>& it)
{
    for (std::size_t i = 0, size = it.size(); i != size; ++i) {
        const std::size_t index = size - 1 - i;
        ++it[index];
        if (it[index] >= v[index].size()) {
            it[index] = 0;
        } else {
            return true;
        }
    }
    return false;
}

void do_job(const std::vector<std::vector<int>>& v,
            const std::vector<std::size_t>& it)
{
    for (std::size_t i = 0; i != it.size(); ++i) {
        // TODO: manage case where v[i] is empty if relevant.
        std::cout << v[i][it[i]] << " ";
    }
    std::cout << std::endl;
}

void iterate(const std::vector<std::vector<int>>& v)
{
    std::vector<std::size_t> it(v.size(), 0u);

    do {
        do_job(v, it);
    } while (increase(v, it));
}

Live Demo