华为OD题目： 取出尽量少的球

华为OD题目： 取出尽量少的球

时间限制: 1s 空间限制: 32MB 限定语言: 不限

题目描述:
某部门开展Family Day开放日活动，其中有个从桶里取球的游戏，游戏规则如下: 有N个容量一样的小桶等距排开，且每个小桶都默认装了数量不等的小球，每个小桶所装的小球数量记录在数组“bucketBalINums中，游戏开始时，要求所有桶的小球总数不能超过SUM，如果小球总数超过SUM，则需对所有的小桶统一设置一个容量最大值maxCapacity，并需将超过容量最大值的小球拿出来，直至小桶里的小球数量小于maxCapacity; 请您根据输入的数据，计算从每个小桶里拿出的小球数量?
限制规则一:
如果所有小桶的小球总和小于SUM，则无需设置容量值，并且无需从小桶中拿球出来，返回结果;
限制规则二:
如果所有小桶的小球总和大于SUM，则需设置容量最大值maxCapacity，并且需从小桶中拿球出来，返回从每个小桶拿出的小球数量组成的数组:

输入描述
第一行输入2个正整数，数字之间使用空格隔开，其中第一个数字表示SUM; 第二个数字表示bucketBalINums数组长度;
第二行输入N个正整数，数字之间使用空格隔开，表示bucketBalINums的每一项;
输出描述:
从每个小桶里拿出的小球数量，并使用一维数组表示
补充说明:
1 <= bucketBallNumsli] <= 10^9
1 <= bucketBallNums.length = N <= 10^5
1 <= maxCapacity <= 10^9
1 <= SUM <= 10^9

示例1
输入:
14 7
2 3 2 5 5 1 4
输出:
[0,1,0,3,3,0,2]

3 3
1 2 3
输出:
[0,1,2]

说明:
小球总数为22，SUM=14，超出范围了，需从小桶取球，
maxCapacity=1，取出球后，桶里剩余小球总和为7，远小于14;
maxCapacitve2，取出小球后，桶里到余小球总和为13，小于最大值;
maxCapacitve3，取出小球后，桶里剩余小球总和为16大于14;
因此maxCapacity=2，每个小桶小球数量大于2的都需要拿出来;

public class My {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String line1 = sc.nextLine();
        String[] strings = line1.split(" ");
        long sum = Long.parseLong(strings[0]);
        int len = Integer.parseInt(strings[1]);

        long[] buckets = new long[len];
        String line = sc.nextLine();
        String[] split = line.split(" ");

        //实际装的球的总量
        long total = 0;
        //数组里单个桶所装的最大值
        long max = 0;
        for (int i = 0; i < len; i++) {
            buckets[i] = Integer.parseInt(split[i]);
            total += buckets[i];
            max = Math.max(max, buckets[i]);
        }
        if (total < sum) {
            System.out.println("[]");
        }

        long minCapacity = sum / len;
        long left = 0;
        long right = max;
        while (left < right) {
            //每个桶的最大容量，因为要向上取整，所以要 left + right + 1
            long mid = (left + right + 1) / 2;
            long tempTotal = 0;
            for (int i = 0; i < len; i++) {
                //如果当前桶的球数量大于容量mid， 取mid，否则取当前数量
                long curNum = Math.min(mid, buckets[i]);
                tempTotal += curNum;
            }
            if (tempTotal <= sum) {
                left = mid;
            }else {
                right = mid - 1;
            }

        }

        StringBuilder sb = new StringBuilder("[");
        for (int i = 0; i < len; i++) {
            if (i != 0) {
                sb.append(",");
            }
            long num = 0;
            if (buckets[i] > left) {
                num = buckets[i] - left;
            }
            sb.append(num);
        }
        sb.append("]");

        System.out.println(sb);


    }

}