题目描述
从 1 ~ n 这 n 个整数中随机选取 m 个数,输出所有可能的选择方案。
输入格式
两个整数 n , m , 在同一行用空格隔开。
输出格式
按照从小到大的顺序输出所有方案,每行 1 个。
首先,同一行内的数升序排列,相邻两个数用一个空格隔开。
其次,对于两个不同的行,对应下标的数一一比较,字典序较小的排在前面(例如1 3 5 7排在1 3 6 8前面)。
数据范围
n
>
0
n > 0
n>0
0
≤
m
≤
n
0 ≤ m ≤ n
0≤m≤n
n
+
(
n
−
m
)
≤
25
n+(n−m)≤25
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>
#include <vector>
using namespace std;
int n,m;
vector<int> result;
void dfs(int index)
{
if(result.size() + n - index < m)
{
return ;
}
if(result.size() == m)
{
for(int i = 0;i < result.size();i++)
{
cout << result[i] << " ";
}
cout << endl;
return ;
}
result.push_back(index + 1);
dfs(index + 1);
result.pop_back();
dfs(index + 1);
}
int main()
{
cin >> n >> m;
dfs(0);
return 0;
}
其二使用函数参数存储,使用位运算表示(当前位为1表示该位表示的数存在)。
#include <iostream>
using namespace std;
int n,m;
void dfs(int index,int sum,int status)
{
if(sum + n - index < m)
{
return ;
}
if(sum == m)
{
for(int i = 0;i < n;i++)
{
if(status >> i & 1)
{
cout << (i + 1) << " ";
}
}
cout << endl;
return ;
}
dfs(index + 1,sum + 1,status | 1 << index);
dfs(index + 1,sum,status);
}
int main()
{
cin >> n >> m;
dfs(0,0,0);
return 0;
}