链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
从 1~n 这 n 个整数中随机选出 m 个,输出所有可能的选择方案。n>0 0≤m≤n n+(n−m)≤25
输入描述:
两个整数n,m。
输出描述:
按照从小到大的顺序输出所有方案,每行1个。
首先,同一行内的数升序排列,相邻两个数用一个空格隔开。其次,对于两个不同的行,对应下标的数一一比较,字典序较小的排在前面(例如1 3 9 12排在1 3 10 11前面)。
示例1
输入
复制5 3
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
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<bits/stdc++.h>
using namespace std;
const int N=50;
int n,m;
int a[N];
bool vis[N];
struct node{
int pos,u,now,x;
};
void dfs(int x,int now,int date){
//0
if(x>n||date+(n-x+1)<m)return ;//超过n位不要,剩下的位数+选的位数小于m结束剪枝
//cout<<u<<endl;
if(date==m){//选了m位
//cout<<u;
for(int i=1;i<=n;++i){if(now>>(i-1)&1)cout<<i<<" ";}//输出
cout<<endl;
return ;
}
dfs(x+1,now+(1<<x),date+1);//选这位
//1
dfs(x+1,now,date);//不选
//2
}
int main(){
cin.tie(0);
cin>>n>>m;
stack<node>st;
st.push({0,0,0,0});
while(st.size()){
auto t=st.top();
st.pop();
if(t.pos==0){
if(t.u>n||t.x+n-t.u<m)continue ;//超过n位不要,剩下的位数+选的位数小于m结束剪枝
if(t.x==m){//选了m位
for(int i=1;i<=n;++i){if(t.now>>(i-1)&1)cout<<i<<" ";}//输出
cout<<endl;
continue;
}
t.pos=1;
st.push(t);
st.push({0,t.u+1,t.now|(1<<t.u),t.x+1});
}
else if(t.pos==1){
t.pos=2;
st.push(t);
st.push({0,t.u+1,t.now,t.x});
}
else continue;
}
}