华为机试真题Java 实现【最大平分数组】【2022.11 Q4】

问题描述

给定一个数组nums，可以将元素分为若干个组，使得每组和相等，求出满足条件的所有分组中，最大的平分组个数。

输入描述：

第一行输入 m

接着输入m个数，表示此数组

数据范围:1<=M<=50, 1<=nums[i]<=50

输出描述：

最大的平分组数个数。

示例1：

输入：

7

4 3 2 3 5 2 1

输出：

4

说明：

可以等分的情况有：

4 个子集（5），（1,4），（2,3），（2,3）

2 个子集（5, 1, 4），（2,3, 2,3）

最大的平分组数个数为4个。

示例2：

输入：

7

5 2 1 5 2 1 5 2 1

输出：

4

说明：

可以等分的情况有：

4 个子集（5，1），（5，1），（5，1），（2，2，2）

2 个子集（5, 1, 5,1），（2,2, 2,5,1）

最大的平分组数个数为4个。

个人解法

不保证通过率

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String s = br.readLine();
        int count = Integer.parseInt(s);
        int[] nums = new int[count];
        String[] split = br.readLine().split(" ");
        int sum = 0;
        for (int i = 0; i < count; i++) {
            nums[i] = Integer.parseInt(split[i]);
            sum += nums[i];
        }
        ArrayList<Integer> list = new ArrayList<>();
        for (int i = 1; i <= sum / 2; i++) {
            if (sum % i == 0) list.add(sum / i);
        }
        Arrays.sort(nums);
        list.sort(Comparator.comparingInt(o -> o));
        for (Integer integer : list) {
            if (integer < nums[count - 1]) continue;
            if (find(nums, integer)) {
                System.out.println(sum / integer);
                return;
            }
        }
    }

    public static boolean find(int[] nums, int per) {
        int len = nums.length;
        boolean[] dp = new boolean[1 << len];
        dp[0] = true;
        int[] temp = new int[1 << len];
        for (int i = 0; i < 1 << len; i++) {
            if (!dp[i]) continue;
            for (int j = 0; j < len; j++) {
                if (temp[i] + nums[j] > per) break;
                if (((i >> j) & 1) == 0) {
                    int next = i | 1 << j;
                    if (!dp[next]) {
                        temp[next] = (temp[i] + nums[j]) % per;
                        dp[next] = true;
                    }
                }
            }
        }
        return dp[(1 << len) - 1];
    }
}