【华为校招】【校招】【Java】分糖果I

题目描述
老师给两个同学分糖果，每袋糖果中的数量不完全一样。一袋糖果只能分给一个人，并且一次性全分完必须。两个人分到的糖果数必须相同。返回两个人分到的糖果数，无法平均分配，返回-1。
输入
第一行输入糖果袋数，范围是[1,100]
第二行是一个数组，每袋糖果数量，范围[1,100]
输出
第一行为每人分到的糖果总数，不能平均分配，输出-1
第二行，第三行为两个同学分配到的每袋糖果具体的糖果数，顺序不限
存在多种分配方式时，输出任意一种可行的方法即可
样例1
输入
1
100
输出
-1

说明
无法平均分配
样例2
输入
5
7 4 5 3 3
输出
11
7 4
5 3 3
思路分析
这道题可以转化为一个0-1背包问题。
可以参考leetcode：416. 分割等和子集

public class DivideCandy {


    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int[] nums = new int[N];
        int sum = 0;
        for (int i=0;i<N;i++){
            nums[i] = sc.nextInt();
            sum+=nums[i];
        }
        if (sum%2==1||N==1){
            System.out.println(-1);
            return;
        }
        int target = sum/2;
        // 表示从下标0-i中任取，放入容量为j的背包，价值总和
        boolean[][] dp = new boolean[N+1][target+1];
        for (int i=0;i<=N;i++){
            dp[i][0] = true;
        }
        for (int i=1;i<=N;i++){
            for (int j=1;j<=target;j++){
                if (j<nums[i-1]){
                    dp[i][j] = dp[i-1][j];
                }else {
                    dp[i][j] = dp[i - 1][j] | dp[i - 1][j - nums[i - 1]];
                }
            }
        }

        // 通过dp可以判断选择前几袋糖果是否可以找到和为target，然后把选择的结果加入list
        List<Integer> list = new ArrayList<>();
        while (target > 0) {
            for (int i = 1; i <= N; i++) {
                if (dp[i][target]) {
                    list.add(nums[i - 1]);
                    target -= nums[i - 1];
                    N = i - 1;
                    break;
                }
            }
        }
        // 后面全部是输出的内容
        Map<Integer, Integer> map = new HashMap<>();  // 存储每袋糖果数量出现的次数，用于处理输出
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
        System.out.println(sum / 2);  // 输出每人分到的糖果总数,target已经变了
        // 第一个同学分配到的每袋糖果的糖果数
        for (int i = 0; i < list.size(); i++) {
            if (i == list.size() - 1) {
                System.out.println(list.get(i));
            } else {
                System.out.print(list.get(i) + " ");
            }
            map.put(nums[i], map.get(nums[i]) - 1);
        }
        // 第二个同学分配到的每袋糖果的糖果数
        List<Integer> list2 = new ArrayList<>();
        for (int num : nums) {
            if (map.get(num) > 0) {
                list2.add(num);
                map.put(num, map.get(num) - 1);
            }
        }
        for (int i = 0; i < list2.size(); i++) {
            if (i == list2.size() - 1) {
                System.out.println(list2.get(i));
            } else {
                System.out.print(list2.get(i) + " ");
            }
        }
    }
}