从N个元素中取k个元素子集的减治法实现

 

对于n个元素的k个分量的所有组合的减治算法如下：

设有6个元素的集合{ 1，2，3，４，５，６　}，现在取其4个分量的所有组合的步骤为：

1.         根据n-k得出从集合中第（k-1）个元素开始取值，即{3},{4},{5},{6}，组成2个分量的所有子集。此时对于元素{3},其子集个数应为3个，即{3,4}{3,5}{3,6}:

                  对于元素{4},其子集个数应为2个，即{4,5}{4,6}:

                  对于元素{5},其子集个数应为1个，即{5,6}

2.         根据第一步得到的2个分量的子集，添加新的元素{2},组成3个分量的所有子集。此时对于元素{2},其子集个数应为第一步中所有子集的个数，即6个，而对于元素{3},{4},{5}，其3个分量的子集个数分别为不以其本身开头的子集个数，分别为：3，1，0。

所以得出的3个分量的所有子集为：

{2,3,4}{2,3,5}{2,3,6}{2,4,5}{2,4,6}{2,5,6}

{3,4,5}{3,4,6}{3,5,6}

{4,5,6}

3.         根据第二步，可知在增加元素{1}之后，元素{1}{2}{3}{4}{5}所对应的4个分量的子集的个数分别为：6+3+1，3+1，1，0，0

所以得到的4个分量的所有子集为：

{1,2,3,4}{1,2,3,5}{1,2,3,6}{1,2,4,5}{1,2,4,6}{1,2,5,6}{1,3,4,5}{1,3,4,6}{1,3,5,6}{1,4,5,6}

{2,3,4,5}{2,3,4,6}{2,3,5,6}{2,4,5,6}

{3,4,5,6}

 

由上述例子可以推出n个元素k个分量的所有组合的减治算法，开始对k个分量进行i=2个元素的组合，然后每一步增加一个分量，做i+1个分量的组合，直到元素个数为n，此时得出k个元素的所有组合了。

C++代码：

#include <iostream>
#include <vector>

using namespace std;

int C( int n, int m)
  {
    if ( n == 0 )
      return 1;
    int res = 1;
    int temp = m - n;
    for( int i = m; i != temp; --i )
      res *= i;
    for( int j = n; j > 1; --j )
      res /= j;
    return res;
  }

int main()
  {
        const int N = 6;
        int Array[] = { 1, 2, 3, 4, 5, 6 };
        //cout << "Please enter the array" << endl;
        //for (int i1 = 0; i1 < N; cin >> Array[i1++]);
        //for( int a = 0; a < N; ++a )
                //cout << Array[a] << endl;
        cout << "Please input K" << endl;
        int K;
        cin >> K;
        int num = N - K + 1;
        vector< vector< int > >  vec( num ), ivec;        //K个元素子集的个数      
        vector< vector< int > >::iterator it = vec.begin();
        for( int i = K - 1; i < N; ++i, ++it )   //先存储第一列数
          {
            ( *it ).push_back( Array[i] );
          }
        cout << vec.size() << endl;
       
        for( int i1 = 1; i1 < K; ++i1 )    //i1为子集里元素的个数
          {
            int keyNum = K - i1 - 1;
            vector< vector<int> >::iterator old_begin = vec.begin(),
                                            old_end = vec.end();
            int index = 0;//辅助控制需要添加新元素的位置
            for( int i2 = 0; keyNum < N - i1; ++keyNum, i2 += C( i1 - 1, N - keyNum - 1 ))
              {
                old_begin = vec.begin() + i2;
                copy( old_begin, old_end, back_inserter( ivec ) );
    
                for( vector< vector< int > >::iterator beg = ivec.begin() + index; beg != ivec.end(); ++beg )
                  {
                    beg->push_back( Array[keyNum] );
                  }
                index += old_end - old_begin;
              }
                  
            vec.clear();
            copy( ivec.begin(), ivec.end(), back_inserter( vec ) );
            ivec.clear();
          }             
       
        cout << "Here is the result !" << endl;
        cout << "There are "
             << vec.size()
             << " subarray"
             << endl;
        for( vector< vector<int> >::iterator it = vec.begin(); it != vec.end(); ++it )
        {
          for( vector<int>::reverse_iterator it1 = (*it).rbegin(); it1 != (*it).rend(); ++it1 )
              cout << *it1;
          ( it + 1 - vec.begin()) % 10 ? cout << " " : cout << endl;
        }

        return 0;
}