7

我需要将用 Python 编写的代码段移植到 C++ 中，但该代码段使用 Python 中的 itertools 组合。

我真正有兴趣移植到 C++ 的行是这一行：

for k in combinations(range(n-i),2*i):

range(n-i)在 Python 中将生成一个列表0 to (n-i) - 1

设 n = 16, i = 5

print range(n-i)

输出：

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

和 python 组合将在该列表中生成所有可能的组合。

例如

print list(combinations(range(n-i),2*i))

输出：

[(0, 1, 2, 3, 4, 5, 6, 7, 8, 9),
(0, 1, 2, 3, 4, 5, 6, 7, 8, 10),
(0, 1, 2, 3, 4, 5, 6, 7, 9, 10),
(0, 1, 2, 3, 4, 5, 6, 8, 9, 10),
(0, 1, 2, 3, 4, 5, 7, 8, 9, 10),
(0, 1, 2, 3, 4, 6, 7, 8, 9, 10),
(0, 1, 2, 3, 5, 6, 7, 8, 9, 10),
(0, 1, 2, 4, 5, 6, 7, 8, 9, 10),
(0, 1, 3, 4, 5, 6, 7, 8, 9, 10),
(0, 2, 3, 4, 5, 6, 7, 8, 9, 10),
(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)]

我想使用std::vector和next_permutation从 C++ 生成类似的输出，但我仍然得到错误的结果。这是我目前的做法：

for(int j = 0; j < n-i; j++) {
        temp_vector.push_back(j); 
}

该代码段相当于range(n-i)Python 中的代码段。

但以下片段：

do {
     myvector.push_back(temp_vector);
} while(next_permutation(temp_vector.begin(),temp_vector.begin()+2*i));
cout<<myvector.size()<<endl;

不等同combinations(range(n-i),2*i))于 Python，我尝试了很多变体，但仍然无法得出我期望的结果。

例如：

设 n = 16 i = 5

Python

>>> print len(list(combinations(range(n-i),2*i)))

11

C++

#include <vector>
#include <iostream>

using namespace std;
int main() {
    vector<int> temp_vector;
    vector< vector<int> > myvector;
    int n = 16, i = 5;
    for(int j = 0; j < n - i; j++) {
            temp_vector.push_back(j);
    }
    do {
            myvector.push_back(temp_vector);
    } while(next_permutation(temp_vector.begin(), temp_vector.begin()+2*i));
    cout<<myvector.size()<<endl;
    return 0;
}

g++ combinations.cpp

./a.out

3628800

任何指导将不胜感激！非常感谢！

4

1 回答 1

4

组合和排列不是一回事。

组合是来自另一个集合的项目子集的无序列表。排列是列表中项目的唯一顺序。

您正在从 11 个事物的列表中生成 10 个事物的所有组合，因此您将获得 11 个结果，每个结果都缺少原始 11 个项目中的一个。

生成每个排列将生成原始 11 个项目的每个唯一顺序。由于本例中的项目都是唯一的，这意味着结果将是 11！列出每个包含所有 11 个项目的位置。但是，您仅从前 10 个项目生成排列，因此您将获得 10 个！列表，其中不包含第 11 项。

您需要找到一种算法来生成组合而不是排列。

没有用于组合的内置算法。std::next_permutation 可用作生成组合的算法的一部分：请参阅在 c++ 中生成组合。

这是组合算法的旧提案草案，包括代码。

于 2012-11-16T19:57:08.740 回答