1

所以，我正在为我的问题寻找一个好的解决方案。

我需要生成（打印）整数列表的所有组合，例如：如果数组包含从 0 到 n-1 的整数，其中 n = 5：

int array[] = {0,1,2,3,4};

组合中整数的顺序并不重要，这意味着 {1,1,3}、{1,3,1} 和 {3,1,1} 实际上是相同的组合，因为它们都包含一个 3 和两个 1。

所以对于上面的数组，长度为 3 的所有组合：

0,0,0 -> the 1st combination
0,0,1
0,0,2
0,0,3
0,0,4
0,1,1 -> this combination is 0,1,1, not 0,1,0 because we already have 0,0,1. 
0,1,2
0,1,3
0,1,4
0,2,2 -> this combination is 0,2,2, not 0,2,0 because we already have 0,0,2. 
0,2,3
.
.
0,4,4
1,1,1 -> this combination is 1,1,1, not 1,0,0 because we already have 0,0,1. 
1,1,2
1,1,3
1,1,4
1,2,2 -> this combination is 1,2,2, not 1,2,0 because we already have 0,1,2.
.
.
4,4,4 -> Last combination

现在我为此编写了代码，但我的问题是：如果数组中的数字不是从 0 到 n-1 的整数，假设数组是这样的

int array[] = {1,3,6,7};

我的代码不适用于这种情况，任何用于解决此问题的算法或代码，，

这是我的代码：

unsigned int next_combination(unsigned int *ar, int n, unsigned int k){
    unsigned int finished = 0;
    unsigned int changed = 0;
    unsigned int i;

    for (i = k - 1; !finished && !changed; i--) {
        if (ar[i] < n - 1) {
            /* Increment this element */
            ar[i]++;
            if (i < k - 1) {
                /* Make the elements after it the same */
                unsigned int j;
                for (j = i + 1; j < k; j++) {
                    ar[j] = ar[j - 1];
                }
            }
            changed = 1;
        }
        finished = i == 0;
    }
    if (!changed) {
        /* Reset to first combination */
        for (i = 0; i < k; i++){
            ar[i] = 0;
        }
    }
    return changed;
}

这是主要的：

int main(){
    unsigned int numbers[] = {0, 0, 0, 0, 0};
    const unsigned int k = 3;
    unsigned int n = 5;

    do{
        for(int i=0 ; i<k ; ++i)
            cout << numbers[i] << " ";
        cout << endl;
    }while (next_combination(numbers, n, k));

    return 0;
}
4

3 回答 3

2

此代码要求“元素池”数组从最小值到最大值排序，没有重复条目。

该函数first_combination将结果数组（“dist”）初始化为第一个组合。在此之后，next_combination在循环中调用，直到它返回 false （就像在您的示例中一样）。“n”和“k”参数已被替换为获取数组大小的模板参数——因此除了结果之外，枚举函数还需要池数组。

#include <iostream>

template<typename T, int N, int K>
void first_combination(const T (&pool)[N], T (&dist)[K]) {
    for(int ki=0; ki<K; ++ki) {
        dist[ki] = pool[0];
    }
}

template<typename T, int N, int K>
bool next_combination(const T (&pool)[N], T (&dist)[K]) {
    int ni = 0;;
    int ki = 0;

    for(;;) {
        const int prev_ni = ni;
        // search the pool for the value in this slot 
        for(ni=0; pool[ni] != dist[ki]; ++ni) {
            if(ni == N) return false; // slot contains a value not found in the pool
        }

        if(++ni < N) break;

        ni = 0;
        dist[ki] = pool[0];
        if(++ki == K) return false;
    }

    int v = pool[ni];

    dist[ki] = v;

    // code below assumes pool[] is sorted
    for(--ki; ki>=0; --ki) {
        if(dist[ki] < v) {
            dist[ki] = v;
        }
        else {
            v = dist[ki];
        }
    }

    return true;
}


template<typename T, int COUNT>
void dumparray( T (&dist)[COUNT]) {
    std::cout << '{';
    for(int i=0; i<COUNT; ++i) {
        if(i) std::cout << ',';
        std::cout << dist[i];
    }
    std::cout << '}' << std::endl;
}

int main(int argc, char* argv[]) {
    const int pool[] = {1,3,6,7};
    int dist[3] = {0};

    first_combination(pool, dist);
    do {
        dumparray(dist);
    } while(next_combination(pool, dist));
    return 0;
}
于 2012-09-21T02:32:15.283 回答
2

如果您有工作代码来生成从0到的所有数字组合n-1，那么这非常简单。你有你的数字数组：

int array[] = {1,3,6,7};

现在， take n = 4，因为数组中有 4 个项目。生成从 0 到 3 的所有组合，并将它们用作数组的索引。您现在可以通过使用该数组的所有索引组合来获得数组值的所有组合。

于 2012-09-20T22:35:02.430 回答
0

所以你需要生成组合的程序（wiki链接）。

在这里你有完整的描述，甚至可以使用算法：http ://compprog.wordpress.com/2007/10/17/generating-combinations-1/

于 2012-09-20T22:34:25.550 回答