1. 程式人生 > >遞迴實現 從n個數中選取m個數的所有組合

遞迴實現 從n個數中選取m個數的所有組合

nn>0個數,從中選取mn>m>0個數,找出所有的組合情況(不分順序)。這樣的組合共有 Cmn=n×(n1)××(nm+1)m!.

一個數組 data 有 n 個元素,從中選取 m 個數的組合 arr,使用遞迴演算法實現是這樣一個過程:
1) 選擇 data的第1個元素為arr的第一個元素,即:arr[0] = data[0];
2) 在data第一個元素之後的其它元素中,選取其餘的 m - 1個數,這是一個上述問題的子問題,遞迴即可。
3) 依次選擇 data的第 2 到 n - m + 1元素作為起始點,再執行1、2步驟。
4) 遞迴演算法過程中的 m = 0 時,輸出 arr 的所有元素。

C++ 程式碼如下:

template <typename T>
void computeAllChoices(std::vector<T> &data, int n, int outLen, int startIndex, int m, int *arr, int arrIndex)
{
    if(m == 0)
    {   
         for (int i = 0; i < outLen; i++)   
         {
            std::cout << arr[i] << "\t";
         }
         std
::cout << std::endl; return; } int endIndex = n - m; for(int i=startIndex; i<=endIndex; i++) { arr[arrIndex] = data[i]; computeAllChoices(data, n, outLen, i+1, m-1, arr, arrIndex+1); } }

測試程式碼如下:

int _tmain(int argc, _TCHAR* argv[])
{
    std
::vector<int> data; for(int i = 0; i < 6; i++) { data.push_back(i+1); } int arr[3]; computeAllChoices(data, data.size(), 3, 0, 3, arr, 0); return 0; }

輸出結果:
這裡寫圖片描述