输入两个整数 n 和 m,从数列1,2,3.......n 中随意取几个数,使其和等于 m ,要求将其中所有的可能组合列出来
输入描述:
每个测试输入包含2个整数,n和m
输出描述:
按每个组合的字典序排列输出,每行输出一种组合
示例1
输入
5 5
输出
1 4 2 3 5
解题思路:
标准dfs解法。
import java.util.*;
public class Main{
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
while(sc.hasNext()){
int n = sc.nextInt();
int m = sc.nextInt();
List<List<Integer>> res = new ArrayList<>();
dfs(n, m, new ArrayList<Integer>(), res, 1);
for(int i=0; i<res.size(); i++){
for(int j=0; j<res.get(i).size() - 1; j++){
System.out.print(res.get(i).get(j) + " ");
}
System.out.println(res.get(i).get(res.get(i).size() - 1));
}
}
}
private static void dfs(int n, int m, List<Integer> list, List<List<Integer>> res, int index){
if(m == 0 && list.size() != 0){
res.add(new ArrayList<Integer>(list));
return;
}
for(int i=index; i<=n; i++){
if(i > m)
return;
list.add(i);
dfs(n, m-i, list, res, i+1);
list.remove(list.size()-1);
}
}
}