正整数 uniq 的组合(顺序不重要)



所以,我一直在寻找一个很好的解决方案来解决我的问题。

我需要生成(打印)整数列表的所有组合,例如:如果数组包含从 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;
}

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

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

现在,取n = 4,因为数组中有 4 个项目。 生成从 0 到 3 的所有组合,并将其用作数组的索引。 现在,通过使用数组中的所有索引组合,您拥有数组值的所有组合。

此代码要求将"元素池"数组从最小值到最大值排序,没有重复的条目。

该函数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;
}

所以你需要程序来生成组合(维基链接)。

在这里,您有完整的描述,甚至是现成的算法:http://compprog.wordpress.com/2007/10/17/generating-combinations-1/

最新更新