从 1∼n这 n 个整数中随机选出 m 个,输出所有可能的选择方案。
输入格式
两个整数 n,m ,在同一行用空格隔开。
输出格式
按照从小到大的顺序输出所有方案,每行 1 个。
首先,同一行内的数升序排列,相邻两个数用一个空格隔开。
其次,对于两个不同的行,对应下标的数一一比较,字典序较小的排在前面(例如 1 3 5 7
排在 1 3 6 8
前面)。
数据范围
n>0
0≤m≤n
n+(n−m)≤25
输入样例:
5 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
代码如下:
#include<iostream>
using namespace std;
const int N=30;
int n,m; // n是总数,m是组合的元素数量
int res[N]; // 存储当前组合的数组
// 深度优先搜索函数
void dfs(int x,int y)
{
// 如果已经选择了m个数字,那么就打印出当前的组合
if(x==m)
{
for(int i=0;i<m;i++) cout<<res[i]<<' ';
cout<<endl;
return;
}
// 遍历所有大于等于y的数字i
for(int i=y;i<n;i++)
{
// 将数字i+1添加到当前组合中
res[x]=i+1;
// 递归地生成剩余的组合
dfs(x+1,i+1);
}
}
int main()
{
cin>>n>>m;
dfs(0,0);
return 0;
}
在这段代码中,dfs(int x, int y)
函数是递归函数,用于生成所有可能的组合。x
是当前深度(已经选择的数字的数量),y
是下一个可以选择的最小数字。
如果x
等于m
,代表已经选择了足够的数字,因此打印出当前的组合并返回。
否则,遍历所有大于等于y
的数字i
,将其添加到当前组合中(res[x] = i+1
),然后递归地调用dfs(x+1, i+1)
。
如下是我自己模拟的DFS回溯过程:
dfs(0,0)
{
for(int i=0;i<n;i++)
{
res[0]=1;
dfs(1,1)
{
for(int i=1;i<n;i++)
{
res[1]=1+1;//i=1
dfs(2,2)
{
for(int i=2;i<n;i++)
{
res[2]=2+1;
dfs(3,3)
{
if(x==3) cout<<1 2 3<<endl;
}
res[2]=3+1;//此时i=3
dfs(3,4)
{
if(x==3) cout<<1 2 4<<endl;
}
res[2]=4+1;//此时i=4
dfs(3,5)
{
if(x==3) cout<<1 2 5<<endl;
}
}
}
//这时dfs(2,2)的循环已结束,现在继续dfs(1,1)
res[1]=2+1;//此时i=2
dfs(2,3)
{
for(int i=3;i<n;i++)
{
res[2]=3+1;
dfs(3,4)
{
if(x==3) cout<<1 3 4<<endl;
}
......依此类推
}
}
}
}
}
}