小明很喜欢选数字,现在小明得到 n 个数,他想得到从 n 个数里面选取 k 个数的所有方案,由于小明的作业很多,现在没有时间解决这个问题,聪明的你可以帮助小明解决这个问题吗?
输入格式
输入包括两行数据,第一行包括两个数 n(1≤n<25) , k(1≤k≤min(n,19)), 表示数列数的个数以及要从数列里面选取数的个数。
第二行 n 个数表示该数列的 n 个数 (0<ai≤100,ai 表示数列里面的数, ai 可能有重复)。
(两个数之间用空格分开)。
输出格式
按字典序输出所有的方案(格式参照样例)
输入样例
4 2
1 1 2 3
输出样例
1 1
1 2
1 3
2 3
数据范围
对于 10% 的数据, 1≤n≤5 , k≤n
对于 50% 的数据, 1≤n≤10 , k≤n
对于 100% 的数据, 1≤n≤24 ,
#include <bits/stdc++.h>
using namespace std;
int n,k,ans[25],a[25],vis[25];
void dfs(int id,int num){//id为当前选择的位置,num为选的k个数
if(num==k){//此方案数选完了
for(int i=0;i<k;i++)
cout<<ans[i]<<' ';//那就输出
cout<<'\n';//换个行
return ;//跳出递归
}
for(int i=id;i<=n;i++){
if(vis[id]==0){
/*if(i-1>=1){//有前一个数字
if(a[i-1]==a[i]&&vis[i-1]==1){
vis[i]=1;//标记
ans[cnt]=a[i];//记到答案数组里
dfs(cnt+1,i+1);
vis[i]=0;//归零
}
if(a[i]!=a[i-1]){//直接选
vis[i]=1;//标记
ans[cnt]=a[i];//记到答案数组里
dfs(cnt+1,i+1);
vis[i]=0;//归零
}
}
else{//第一个数字
vis[i]=1;//标记
ans[cnt]=a[i];//记到答案数组里
dfs(cnt+1,i+1);
vis[i]=0;//归零
}
*/
if(i>1&&vis[i-1]==0&&a[i]==a[i-1])//首先判断i>1,也就是此数在数组里不是第一项,然后判断标记数组的前一项是0(未被标记的),最后判断答案数组里存的此项和上一项相等
continue;//那就不继续这次选数了,是和上一次重合的
ans[num]=a[i];//先存上 (下面这四行代码加不加else都一样)
vis[i]=1;//标记
dfs(i+1,num+1);//递归选数
vis[i]=0;//归零
}
}
}
int main() {
cin>>n>>k;
for(int i=1;i<=n;i++){
cin>>a[i];//输入
}
sort(a+1,a+1+n);//先把输入来的数排个序,这样相同的数就都挨在一起了
dfs(1,0);
return 0;
}