华为OD机试-开放日活动-2022Q4 A卷-Py/Java/JS

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

限制规则一:
所有所有小桶的小球总和小于 SUM，则无需设置容量值，并且无需从小桶中拿球出来，返回结果[];

限制规则二:
如果所有小桶的小球总和大于 SUM，则需设置容量最大值 maxcapacity并且需从小桶中拿球出来，返回从每个小桶拿出的小球数量组成的数组;
输入描述

第一行输入2个正整数，数字之间使用空格隔开，其中第一个数字表示 SUM ，第二个数字表示 bucketBallNums 数组长度:第二行输入N个正整数，数字之间使用空格隔开，表示 bucketBallNums 的每一项:
输出描述

数组剩余小球。

示例一

输入

14 7
2 3 2 5 5 1 4

输出
[0,1,0,3,3,0,2]
说明
小球总数为 22，sum=14，超出范围了，需从小桶取球，
maxCapacity=1，取出球后，桶里剩余小球总和为7，远小于 14 maxCapacity=2，取出球后，桶里剩余小球总和为13
maxCapacity=3，取出球后，桶里剩余小球总和为 16，大于14因此 maxCapacity为2，每个小桶小球数量大于2的都需要拿出来;

示例二

输入
3 3
1 2 3
输出
[0,1,2]
说明
小球总数为6，SUM=3，超出范围了，需从小桶取球 maxCapacity=1，则小球总数为3从0号桶取出0个球，从1号桶取出1个球，从2号桶取出2个球;

示例三

输入
6 2
3 2
输出

[]

说明
小球总数为5，SUM=6，在范围内，无需从小桶取球;
备注

1<= bucketBallNums[i] <= 10^9
1 <= bucketBallNums.length = N <= 10^5

1<= maxCapacity<= 10^9

1<= SUM <= 10^9

Java 代码

import java.util.Scanner;
import java.util.*;
import java.util.stream.Collectors;
import java.math.BigInteger;
import java.util.stream.IntStream;
 
class Main {
	public static void main(String[] args) {
        // 处理输入
        Scanner in = new Scanner(System.in);
        int sum = in.nextInt();
        int nums = in.nextInt();
        int[] balls = new int[nums];
        int[] org_balls = new int[nums];
        for (int i = 0; i < nums; i++) {
            balls[i] = in.nextInt();
            org_balls[i] = balls[i];
        }
 
        int total =  Arrays.stream(balls).sum();
        if (total <= sum) {
            System.out.println("[]");
            return;
        }
        Arrays.sort(balls);
 
        //二分法初始化
        int left = sum / nums;
        int right = balls[nums - 1];
        int[] result = new int[nums];
        for (int i = 0; i < nums; i++) {
            result[i] = org_balls[i] > left ? org_balls[i] - left : 0;
        }
 
        while (right > left+1){
            int mid = (right + left) / 2;
        
            int[] tmp = new int[nums];
            int temp_total = total;
            for (int i = 0; i < balls.length; i++) {
                int r = org_balls[i] > mid? org_balls[i] - mid: 0;
                temp_total -= r;
                tmp[i] = r;
            }
 
            if (temp_total > sum)
                right = mid;
            else if (temp_total < sum){
                left = mid;
                result = tmp;
            }
            else{
                result = tmp;
                break;
            }
        }
        
            
        System.out.println(Arrays.toString(result));
    }
 
}

Python代码

import functools
import collections
import math
from itertools import combinations
from re import match
 
 
 
# 处理输入
params = [int(x) for x in input().split(" ")]
total = params[0]
N = params[1]
balls = [int(x) for x in input().split(" ")]
origin_balls = []
for i in balls:
    origin_balls.append(i)
single_total = sum(balls)
 
 
 
if(single_total <= total):
    print([])
else:
    #二分法初始化
    left = int(total / N)
    right = max(balls)
    result = [x - left if x > left else 0 for x in origin_balls]
    while (right > left+1):
        mid = int((right + left) / 2)
    
        temp = []
        for i in range(N):
            if(balls[i] > mid):
                temp.append(origin_balls[i] - mid)
            else:
                temp.append(0)
    
        if (single_total - sum(temp)>total):
            right = mid
        elif (single_total -sum(temp)

JS代码

function main(sum, nums, balls ) {
    let total = eval(balls.join("+"))
    if (total <= sum) {
        console.log("[]")
        return
    }
    let org_balls = []
    for (let i=0;i left ? org_balls[i] - left : 0;
    }
 
    while (right > left+1){
        let mid = (right + left) / 2
    
        let tmp = new Array(nums)
        let temp_total = total
        for (let i = 0; i < balls.length; i++) {
            let r = org_balls[i] > mid? org_balls[i] - mid : 0;
            temp_total -= r;
            tmp[i] = r;
        }
 
        if (temp_total > sum)
            right = mid;
        else if (temp_total < sum){
            left = mid;
            result = tmp;
        }
        else{
            result = tmp;
            break;
        }
    }
 
    
    console.log(result)
 
}
 
 
main(3,3,[2,3,1])