Java实现 蓝桥杯VIP 算法训练 和为T

问题描述

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

输入格式

第一行一个正整数n，表示整数集内元素的个数。

第二行n个整数，用空格隔开。

第三行一个整数T，表示要达到的和。

输出格式

输出有若干行，每行输出一组解，即所选取的数字，按照输入中的顺序排列。

若有多组解，优先输出不包含第n个整数的；若都包含或都不包含，优先输出不包含第n-1个整数的，依次类推。

最后一行输出总方案数。

样例输入

5

-7 -3 -2 5 9

样例输出

-3 -2 5

-7 -2 9

2

数据规模和约定

1<=n<=22

T<=maxlongint

集合中任意元素的和都不超过long的范围

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


public class 和为T {
	  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(), 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 out = new Stack<Integer>();
	    
	    private static void dfs(int pos, int tempT, LinkedList 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 it) {
	        StringBuilder result = new StringBuilder();
	        while (it.hasNext()) {
	            result.append(it.next() + " ");
	        }
	        return result.toString();
	    }

}