【华为机试真题详解 Java实现】数组限制数【2023 Q1 考试题 A卷 200分】

题目描述

【数组限制数】

给定一个数组，可以给数组添加一个限制使得数组中的每一个数和限制取min，要使数组的总和不超过 total，
求限制最高可以取多少，如果不需要限制则返回-1。

输入

第一行包含两个整数n，m(1 ≤ n,m,ai ≤ 100000)，分别代表数组的长度和不能超过的数组的总和。

第二行包含n个整数，代表数组的每个元素。

解题思路

这道题可以使用二分查找来解决。具体地，我们可以二分枚举限制的取值范围，每次假设限制的取值为mid，然后判断在这个限制下数组的总和是否小于等于total。如果小于等于total，则我们可以尝试增加限制的取值，否则我们需要降低限制的取值。

具体实现时，我们可以使用一个变量sum来记录数组中小于等于限制的元素的总和，每次遍历数组，如果当前元素小于等于限制，那么我们就将这个元素加入sum中。最后，如果sum小于等于total，那么说明限制取值为mid是可行的，我们需要将限制的取值范围缩小到[mid+1, right]；否则，限制的取值范围应该缩小到[left, mid-1]。
时间复杂度为 O(n\log W)，其中n为数组的长度，W为数组中的最大值。

参考代码

public static int getMinLimit(int[] nums, int total) {
    int n = nums.length;
    int left = 0, right = Arrays.stream(nums).max().getAsInt();
    int ans = -1;
    while (left <= right) {
        int mid = (left + right) / 2;
        int sum = 0;
        for (int i = 0; i < n; i++) {
            sum += Math.min(nums[i], mid);
        }
        if (sum <= total) {
            ans = mid;
            left = mid + 1;
        } else {
            right = mid - 1;
        }
    }
    return ans;
}