最大平分数组（华为OD测试题）

题目描述

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

输入

第一行输入 m 

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

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

输出

最大的平分组数个数。

样例输入 复制

7
4 3 2 3 5 2 1

样例输出 复制

4

代码展示

import java.util.Arrays;
import java.util.Scanner;

public class Main1 {
    public static void main(String[] args) {
        System.out.println("请输入数组长度（数组间空格隔开）：");
        Scanner sc = new Scanner(System.in);
        int m = sc.nextInt();
        int[] nums = new int[m];
        for(int i=0;i<m;i++){
            nums[i] = sc.nextInt();
        }
        int result = getResult(nums, m);
        System.out.println(result);
    }

    public static int getResult(int[] nums, int m) {
        Arrays.sort(nums);
        int sum = 0;
        for (int ele : nums) {
            sum += ele;
        }
        int target = sum / m;
        while (m >= 1) {
            if (canPartitionMSubsets(nums.clone(), sum, target, m)) {
                return m;
            }
            m--;
        }
        return 0;
    }

    public static boolean canPartitionMSubsets(int[] nums, int sum, int target, int m) {
        if (sum % m != 0) { // 如果总和不能分为m份，则不可能有解
            return false;
        }
        int subSum = sum / m;
        if (subSum < nums[nums.length - 1]) { // 如果最大元素比每份的平均值还要大，则没有解
            return false;
        }
        int index = nums.length - 1;
        while (index >= 0 && nums[index] == subSum) { // 将数组中所有等于平均值的元素都从数组中去掉
            index--;
            m--;
        }
        int[] buckets = new int[m];
        return partition(nums, index, buckets, subSum);
    }

    public static boolean partition(int[] nums, int index, int[] buckets, int target) {
        if (index < 0) { // 如果所有元素都放入桶中，则找到了一种解法
            return true;
        }
        int select = nums[index];
        for (int i = 0; i < buckets.length; i++) {
            if (i > 0 && buckets[i] == buckets[i - 1]) { // 避免重复
                continue;
            }
            if (select + buckets[i] <= target) { // 如果当前元素可以加入当前桶中
                buckets[i] += select;
                if (partition(nums, index - 1, buckets, target)) { // 继续处理下一个元素
                    return true;
                }

                buckets[i] -= select;
            }
        }
        return false;
    }
}