從數組中取出n個元素的所有組合(遞歸實現)


今天在做POJ 1753時,需要枚舉一個數組中所有組合。之前也遇到過類似的問題,如求從n個數組任意選取一個元素的所有組合都是想起來比較簡單,但是設計成算法卻頗費周折。
如數組為{1, 2, 3, 4, 5, 6},那么從它中取出3個元素的組合有哪些,取出4個元素的組合呢?
比如取3個元素的組合,我們的思維是:
取1、2,然后再分別取3,4,5,6;
取1、3,然后再分別取4,5,6;
......
取2、3,然后再分別取4,5,5;
......
這樣按順序來,就可以保證完全沒有重復。

這種順序思維給我們的啟示便是這個問題可以用遞歸來實現,但是僅從上述描述來看,卻無法下手。
我們可以稍作改變:
1.先從數組中A取出一個元素,然后再從余下的元素B中取出一個元素,然后又在余下的元素C中取出一個元素
2.按照數組索引從小到大依次取,避免重復

依照上面的遞歸原則,我們可以設計如下的算法,按照索引從小到大遍歷:

 1 //arr為原始數組
 2 //start為遍歷起始位置
 3 //result保存結果,為一維數組
 4 //count為result數組的索引值,起輔助作用
 5 //NUM為要選取的元素個數
 6 //arr_len為原始數組的長度,為定值
 7 void combine_increase(int* arr, int start, int* result, int count, const int NUM, const int arr_len)
 8 {
 9   int i = 0;
10   for (i = start; i < arr_len + 1 - count; i++)
11   {
12     result[count - 1] = i;
13     if (count - 1 == 0)
14     {
15       int j;
16       for (j = NUM - 1; j >= 0; j--)
17         printf("%d\t",arr[result[j]]);
18       printf("\n");
19     }
20     else
21       combine_increase(arr, i + 1, result, count - 1, NUM, arr_len);
22   }
23 }

當然,我們也可以按照索引從大到小進行遍歷:

 1 //arr為原始數組
 2 //start為遍歷起始位置
 3 //result保存結果,為一維數組
 4 //count為result數組的索引值,起輔助作用
 5 //NUM為要選取的元素個數
 6 void combine_decrease(int* arr, int start, int* result, int count, const int NUM)
 7 {
 8   int i;
 9   for (i = start; i >=count; i--)
10   {
11     result[count - 1] = i - 1;
12     if (count > 1)
13     {
14       combine_decrease(arr, i - 1, result, count - 1, NUM);
15     }
16     else
17     {
18       int j;
19       for (j = NUM - 1; j >=0; j--)
20     printf("%d\t",arr[result[j]]);
21       printf("\n");
22     }
23   }
24 }

測試代碼:

 1 #include <stdio.h>
 2 
 3 int main()
 4 {
 5   int arr[] = {1, 2, 3, 4, 5, 6};
 6   int num = 4;
 7   int result[num];
 8 
 9   combine_increase(arr, 0, result, num, num, sizeof(arr)/sizeof(int));
10   printf("分界線\n");
11   combine_decrease(arr, sizeof(arr)/sizeof(int), result, num, num);
12   return 0;
13 }

輸出結果為:

 1 1    2    3    4    
 2 1    2    3    5    
 3 1    2    3    6    
 4 1    2    4    5    
 5 1    2    4    6    
 6 1    2    5    6    
 7 1    3    4    5    
 8 1    3    4    6    
 9 1    3    5    6    
10 1    4    5    6    
11 2    3    4    5    
12 2    3    4    6    
13 2    3    5    6    
14 2    4    5    6    
15 3    4    5    6    
16 分界線
17 6    5    4    3    
18 6    5    4    2    
19 6    5    4    1    
20 6    5    3    2    
21 6    5    3    1    
22 6    5    2    1    
23 6    4    3    2    
24 6    4    3    1    
25 6    4    2    1    
26 6    3    2    1    
27 5    4    3    2    
28 5    4    3    1    
29 5    4    2    1    
30 5    3    2    1    
31 4    3    2    1    


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM