递归实现组合型枚举
从 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
这一条与上一条并没有太大的差别,唯一要注意的就是递归的终止条件以及答案的顺序问题。
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int m=scanner.nextInt();
int[] a = new int[n];
for (int i = 0; i < n; i++) {
a[i] = i + 1;
}
List<List<Integer>> res=new ArrayList<>();
List<Integer> path = new ArrayList<>();
dfs(res,path,a,0,m);
for (List<Integer> ele:res){
for (int i=0;i<ele.size();i++){
System.out.print(ele.get(i)+" ");
}
System.out.println();
}
}
private static void dfs(List<List<Integer>> res, List<Integer> path, int[] a, int i, int m) {
if (path.size()==m){
res.add(new ArrayList<>(path));
return;
}else if (i==a.length){
return;
}
for (int j=0;j<2;j++){
//j==0不选,j==1选择
if (j==1){
dfs(res, path, a, i+1, m);
}else {
path.add(a[i]);
dfs(res, path,a,i+1, m);
//回溯
path.remove(path.size()-1);
}
}
}
}