题目描述
从 1~n 这 n 个整数中随机选出 m 个,输出所有可能的选择方案。 n > 0, 0 ≤ m ≤ n, n + ( n − m ) ≤ 25。
输入描述:
两个整数n,m。
输出描述:
按照从小到大的顺序输出所有方案,每行1个。
首先,同一行内的数升序排列,相邻两个数用一个空格隔开。其次,对于两个不同的行,对应下标的数一一比较,字典序较小的排在前面(例如1 3 9 12排在1 3 10 11前面)。
思路:
这道题与 递归实现组合型枚举 这道题完全一致,只是使用递归与不使用的区别,然而这两题使用不使用都能AC,建议参考我写的 递归实现组合型枚举--题解 这道题的代码是可以依据递归的深搜化为3步进行
void dfs(int u, int sum, int state) {
//0:
if (sum + n - u < m) return;
if (sum == m) {
for (int i = 0; i < n; i++)
if (state >> i & 1)
cout << i + 1 << " ";
cout << endl;
return;
}
dfs(u + 1, sum + 1, state | (1 << u));
//1:
dfs(u + 1, sum, state);
//2:
}这里就不多解释了
完整C++版AC代码(非递归)
#include <iostream>
#include <stack>
using namespace std;
struct State {
int pos, u, sum, state;//pos是你遍历到程序的哪一步了
};
int main() {
int n, m;
cin >> n >> m;
stack<State> stk;//栈记录好你遍历分支的哪一个部分
stk.push({ 0,0,0,0 });
while (stk.size()) {
auto t = stk.top();
stk.pop();
if (t.pos == 0) {
if (t.sum + n - t.u < m) continue;
if (t.sum == m) {
for (int i = 0; i < n; i++)
if (t.state >> i & 1)
cout << i + 1 << " ";
cout << endl;
continue;
}
t.pos = 1;
stk.push(t);
stk.push({ 0,t.u + 1,t.sum + 1,t.state | 1 << t.u });
continue;
}
if (t.pos == 1) {
t.pos = 2;
stk.push(t);
stk.push({ 0,t.u + 1,t.sum,t.state });
continue;
}
else {
continue;
}
}
return 0;
}
京公网安备 11010502036488号