蓝桥杯 ALGO-115 算法训练 和为T Java版

问题描述

从一个大小为n的整数集中选取一些元素，使得它们的和等于给定的值T。每个元素限选一次，不能一个都不选。

输入格式

第一行一个正整数n，表示整数集内元素的个数。
第二行n个整数，用空格隔开。
第三行一个整数T，表示要达到的和。

输出格式

输出有若干行，每行输出一组解，即所选取的数字，按照输入中的顺序排列。
若有多组解，优先输出不包含第n个整数的；若都包含或都不包含，优先输出不包含第n-1个整数的，依次类推。
最后一行输出总方案数。

样例输入

5
-7 -3 -2 5 9
0

样例输出

-3 -2 5
-7 -2 9
2

数据规模和约定

1<=n<=22
T<=maxlongint
集合中任意元素的和都不超过long的范围

【分析】一道深搜的题目，难点在于输出多组解的顺序，为了方便解决输出多组解的顺序的这个问题，需要从后想前搜索，然后把结果逆序输出就可以得到结果～

如例子，从9开始搜索，得到9 -2 -7，然后把它的顺序反一下输出就得到了一组解，而且这个解在另一组解5 -2 -3的前面先得到，所以最后把所有的解按逆序输出就得到了结果～

import java.util.Iterator;
import java.util.LinkedList;
import java.util.Scanner;
import java.util.Stack;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        isVisited = new boolean[n];
        nums = new int[n];
        for (int i = 0; i < n; i++) {
            nums[i] = sc.nextInt();
        }
        t = sc.nextInt();
        sc.close();
        dfs(n - 1, 0, new LinkedList<Integer>(), true);
        while (!out.isEmpty()) {
            System.out.println(out.pop());
        }
        System.out.println(cnt);
    }

    private static int cnt = 0, n = 0, t = 0;
    private static boolean[] isVisited;
    private static int[] nums;
    private static Stack<String> out = new Stack<>();

    private static void dfs(int pos, int tempT, LinkedList<Integer> selectedNums, boolean isNone) {
        for (int i = pos; i >= 0; i--) {
            if (!isVisited[i]) {
                isVisited[i] = true;
                selectedNums.push(nums[i]);
                dfs(i - 1, tempT + nums[i], selectedNums, false);
                selectedNums.pop();
                isVisited[i] = false;
            }
        }
        if (tempT == t && !isNone) {
            out.push(toReulstString(selectedNums.iterator()));
            cnt++;
        }
    }

    private static String toReulstString(Iterator<Integer> it) {
        StringBuilder result = new StringBuilder();
        while (it.hasNext()) {
            result.append(it.next() + " ");
        }
        return result.toString();
    }
}