问题 B: 【递归入门】组合的输出
时间限制: 1 Sec 内存限制: 128 MB
提交: 194 解决: 117
[提交][状态][讨论版][命题人:外部导入]
题目描述
排列与组合是常用的数学方法,其中组合就是从n个元素中抽出r个元素(不分顺序且r < = n),我们可以简单地将n个元素理解为自然数1,2,…,n,从中任取r个数。
现要求你不用递归的方法输出所有组合。
例如n = 5 ,r = 3 ,所有组合为:
1 2 3
1 2 4
1 2 5
1 3 4
1 3 5
1 4 5
2 3 4
2 3 5
2 4 5
3 4 5
输入
一行两个自然数n、r ( 1 < n < 21,1 < = r < = n )。
输出
所有的组合,每一个组合占一行且其中的元素按由小到大的顺序排列,所有的组合也按字典顺序。
/*
递归写法:(dfs内亦可用循环):
v[num]=index;
for(int i = num + 1; i <= n; i++)
DFS(i, index + 1);
且此时不需要判定if(index==n) return;
*/
#include<iostream>
#include<vector>
using namespace std;
int n, r;
vector<int> v;
void DFS(int index, int num) {//当前处理下标为index的数,已选num个数
//递归边界:
if (num == r) {//选够数字return
for (int i = 0; i < r; i++)
cout << v[i] << " ";
return;
}
if (index == n) return;//超出范围return
//递归核心:
v.push_back(index + 1); //若选择下标为index的index+1
DFS(index + 1, num + 1); //继续选择下一个(直到结束)
v.pop_back(); //已尝试过index,返回状态
DFS(index + 1, num); //若不选择下标为index的index+1
}
int main() {
while (cin >> n >> r) {
DFS(0, 0);
}
return 0;
}
/*STL algorithm中prev_permutation函数:
a存储总排列,b存储数字。例如 5 3:
则a数组开始为11100,b数组为12345
开始输出123,a数组前一个排列为11010,输出124
以此类推直到00111输出345结束
#include <cstdio>
#include <algorithm>
using namespace std;
int main(){
int n, m;
scanf("%d %d", &n, &m);
int a[25] = { 0 }, b[25] = { 0 };
for (int i = 0; i < m; i++)
a[i] = 1;
for (int i = 0; i < n; i++)
b[i] = i + 1;
for (int i = 0; i < m; i++) {
printf("%d ", b[i]);
}
printf("\n");
while (prev_permutation(a, a + n)){
for (int i = 0; i < n; i++)
if (a[i])
printf("%d ", b[i]);
printf("\n");
}
}
*/