输入两个整数n和m,从数列1,2,3.......n中随意取几个数,使其和等于m,要求将其中所有的可能组合列出来
输入描述:
每个测试输入包含2个整数,n和m
输出描述:
按每个组合的字典序排列输出,每行输出一种组合
示例1
输入
55
输出
14
23
5
题目的意思:
从1到n中随机选取几个数,让其和等于m根据这句话,选出来的数不能重复,但是选取的个数任意
所以可以利用深度优先搜索【DFS】+回溯来解决
其中深度优先搜索算法用来寻找所有可能的组合,而回溯是用来筛选出所有可能组合中的可行解【如果发现某候选解不是可行解,直接丢弃】
注:由于选出来的数不能重复,所以在进行深搜时,下一次深搜应是从当前元素的下一个元素继续往后看
步骤:
1.从第一个元素开始相加
2.让局部和继续累加候选的剩余值
3.局部和等于目标值,保存组合,向上回退,寻找其它组合
//DFS+回溯
public class GetSum {
private static List<ArrayList<Integer>> res = new ArrayList<>();
private static List<Integer> list = new ArrayList<>();
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
while (scanner.hasNext()) {
int n = scanner.nextInt();
int m = scanner.nextInt();
baskTrace(1, 0, m, n);
//打印最终结果
for (ArrayList<Integer> l : res) {
for (int i = 0; i < l.size(); i++) {
System.out.print(l.get(i));
if (i != l.size() - 1) {
System.out.print(" ");
}
}
System.out.println();
}
}
}
private static void baskTrace(int beginIndex, int sum, int target, int n) {
//边界条件,如果大于等于目标值,就结束【因为此时的和已经大于或等于目标值了,再继续深搜只会让和更加大于目标值,而这样的组合明显不合题意】
if (sum >= target) {
//如果等于目标值,表示找到一种组合,放入最终集合中
if (sum == target) {
res.add(new ArrayList<>(list));
}
return;
}
for (int i = beginIndex; i <= n; i++) {
//如果当前值比目标值大,直接跳过
if (i > target) {
continue;
}
list.add(i);
//从当前元素的下一个元素开始往后看【因为元素不可以重复】
baskTrace(i + 1, sum + i, target, n); //区别:beginIndex为当前元素的下一个元素!!!
//回溯,向上回退,寻找下一种组合
list.remove(list.size() - 1);
}
}
// n=5 m=5
//数列:1,2,3,4,5
// 从1开始深搜:
// 1<5 -> 1+2<5 -> 1+2+3>5【丢弃3,且不再 在1+2+3的基础上继续深搜】 -> 1+2+4>5【丢弃4,且不再 在1+2+4的基础上继续深搜】 -> 1+2+5>5【丢弃5,且不再 在1+2+5的基础上继续深搜】 -> 丢弃2【1+2的深搜完毕】
// -> 1+3<5 -> 1+3+4>5【丢弃4,且不再 在1+3+4的基础上继续深搜】 -> 1+3+5>5【丢弃5,且不再 在1+3+5的基础上继续深搜】 -> 丢弃3【1+3的深搜完毕】
// -> 1+4==5【找到一种组合,存入最终结果】 ->丢弃4【1+4的深搜完毕】
// -> 1+5>5【丢弃5,1+5的深搜完毕】
// -> 丢弃1【1的深搜完毕】
// 从2开始深搜:
// 2<5 -> 2+3==5【找到一种组合,存入最终结果】 -> 丢弃3【2+3的深搜完毕】
// -> 2+4>5【丢弃4,2+4的深搜完毕】
// -> 2+5>5【丢弃5,2+5的深搜完毕】
// -> 丢弃2【2的深搜完毕】
// 从3开始深搜:
// 3<5 -> 3+4>5【丢弃4,3+4的深搜完毕】
// -> 3+5>5【丢弃5,3+5的深搜完毕】
// -> 丢弃3【3的深搜完毕】
// 从4开始深搜:
// 4<5 -> 4+5>5【丢弃5,4+5的深搜完毕】
// -> 丢弃4【4的深搜完毕】
// 从5开始深搜:
// 5==5【找到一种组合,存入最终结果】
// -> 丢弃5【5的深搜完毕】
//数列1到5已遍历完毕
//最终结果:[[1,4],[2,3],[5]]
}