求和_牛客网

输入两个整数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]]
}