Java 蓝桥杯 算法 和为T

题目描述

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

输入
输入描述:
　　第一行一个正整数n，表示整数集内元素的个数。
　　第二行n个整数，用空格隔开。
　　第三行一个整数T，表示要达到的和。
输入样例:
5
-7 -3 -2 5 9
0

输出

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

HINT:时间限制：1.0s 内存限制：256.0MB
　　1<=n<=22
　　T<=maxlongint
　　集合中任意元素的和都不超过long的范围

解题思路

可以直接遍历数组中每个数加其余各数之后在数组中是否有与其和相等的数就行了

代码

package com.company;

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

public class Main {
    static int cnt = 0;
    static int t = 0;
    static boolean[] sum;
    static int[] nums;
    static final Stack out = new Stack<Integer>();
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();                       //输入数据
        sum = 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 void dfs(int pos, int tempT, LinkedList<Object> list, boolean isNone) {
        for (int i = pos; i >= 0; i--) {
            if (!sum[i]) {
                sum[i] = true;
                list.push(nums[i]);
                dfs(i - 1, tempT + nums[i], list, false);//遍历数组
                list.pop();//判断是否相等
                sum[i] = false;
            }
        }
        if (tempT == t && !isNone) {
            out.push(ber(list.iterator()));             //得到满足的数组
            cnt++;                                      //个数加一
        }
    }

    private static String ber(Iterator<Object> it) {
        StringBuilder result = new StringBuilder();
        while (it.hasNext()) {
            result.append(it.next()).append(" ");//得到满足的数组元素
        }
        return result.toString();
    }
}