Question

我需要一种方法来生成数组长度为k的所有唯一子集，类似于python中的itertools.combinations（）。我试图获得这些用于关联挖掘的集合，直到支持小于2.任何通过2的集合都被证明是令人头疼的。此外，不能使用任何STL功能。

任何见解都将受到赞赏。

最近我能找到我的问题所在：Generate all subsets of size k (containing k elements) in Python

P.S集合（数组）中的每个元素都是整数。

在main中：

int temp = 0;
int setsize = 2;
int **testarray = subsets(itemset,itemsetSize,setsize);
for (int i = 0; i < 1300; i++)
{
    for(int j = 0; j < 10; j++)
    {
        cout<<testarray[i][j];
    }
}
cout<<endl;

在函数中：

int** subsets (int inputArray[],int arraySize, int k)
{
    size_t n = arraySize;

    int **resultArray = 0;
    resultArray = new int *[1000];

    size_t i = (1 << k) -1;

    while( !(i >> n) ) 
    {
        int pushBack = 0;
        int pushBack2 = 0;
        int v[500];
        for (size_t j = 0; j<n; j++)
        {
            if (i & (1 << j))
            {
                v[pushBack] = inputArray[j];
                pushBack++;
            }
        }
        resultArray[pushBack2] = v;
        pushBack2++;
        i = (i+(i&(-i)))|(((i^(i+(i&(-i))))>>2)/(i&(-i)));          
    }
    return resultArray;
}

Answer 1

#include <vector>
#include <iostream>
using namespace std;

vector<vector<int>> subsets(const vector<int>& input, int k)
{
    size_t n = input.size();

    vector<vector<int>> result;

    size_t i = (1 << k) - 1;

    while( !(i >> n) ) 
    {
        vector<int> v;

        for (size_t j = 0; j < n; j++)
           if (i & (1 << j))
                v.push_back(input[j]);

        result.push_back(v);

        i = (i+(i&(-i)))|(((i^(i+(i&(-i))))>>2)/(i&(-i))); 
    }

    return result;
}

int main()
{
    auto out = subsets({5,2,8,14,3,9,11}, 3);
    for (auto x : out)
    {
        for (auto y : x)
            cout << y << " ";
        cout << endl;
    }
}

输出：

如何生成数组c ++的长度为k的所有唯一子集

1 个答案: