华为OD机试: 39 查找充电设备组合

package a_od_test;

import java.util.*;

/*
查找充电设备组合
时间限制：5s空间限制：256MB限定语言：不限

题目描述：
某个充电站，可提供n个充电设备，每个充电设备均有对应的輸出功率。任意个充电
设备组合的输出功率总和，均构成功率集合P的1个元素。功率集合P的最优元秦，表
示最接近充电站最大输出功率p_max的元素。

输入描述：
输入为3行：
第1行为充电设备个数n.
第2行为每个充电设备的输出功率。
第3行为充电站最大输出功率p_maxₒ

输出描述：
功率隼合P的最优元表
补充说明：
1 充电设备个数n>0
2.最优元素必须小于或等于充电站最大输出功率p_maxₒ

示例1:
输入:
4
50 20 20 60
90
输出:
90
说明:
当充电设备输出功率为50,20,20组合时,其充电功率总和为90,最接近充电站最大充电输出功率,
因此最优元素为90

示例2:
输入:
2
50 40
30
输出:
0
说明:
所有率电设备的输出功率组合,均大于充电站最大充电功率30,此时最优元素为0

解题思路:
算法一:
使用回溯法对所有充电设备进行求和
算法二:
通过双层for循环求出所有符合要求的组合的输出总功率
 */
public class Main39 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int num = Integer.parseInt(sc.nextLine());
        String[] split = sc.nextLine().split(" ");
        int max = Integer.parseInt(sc.nextLine());

        int[] ints = new int[num];
        for (int i = 0; i < num; i++) {
            ints[i] = Integer.parseInt(split[i]);
        }

        Set<Integer> set = new TreeSet<Integer>();
        for (int i = 0; i < num; i++) {
            set.add(ints[i]);
            int tmp = ints[i];
            for (int j = i + 1; j < num; j++) {
                tmp = tmp + ints[j];
                set.add(tmp);
            }
        }

        Iterator<Integer> iterator = set.iterator();
        int result = 0;
        while (iterator.hasNext()) {
            Integer next = iterator.next();
            if (next > max) {
                break;
            }
            result = next;
        }

        System.out.println(result);

    }
}