LeetCode刷题笔记 字节每日打卡 分割数组的最大值

给定一个非负整数数组 nums 和一个整数 m ，你需要将这个数组分成 m 个非空的连续子数组。

设计一个算法使得这 m 个子数组各自和的最大值最小。

示例 1：

输入：nums = [7,2,5,10,8], m = 2
输出：18
解释：
一共有四种方法将 nums 分割为 2 个子数组。 
其中最好的方式是将其分为 [7,2,5] 和 [10,8] 。
因为此时这两个子数组各自的和的最大值为18，在所有情况中最小。

参考： 力扣

---

 二分法

结果在[单个最大，数组和]之间

mid作为check的最大值

通过check说明分段数可以满足最大值还可以缩小 right==mid

不通过则说明分段数无法满足需要增大 left==mid+1

check函数

输入参数：数组nums，每段不能超过的最大值x，段数m

参数：sum当前分割子数组的和 cnt表示已分割的子数组数量（包括当前已分割）

遍历每个字符

当前段和超过最大，分段，sum该进下一段

否则sum+=nums[i]，继续该段

当前分段数<=m说明当前合格，该分段数在当前最大值下可以满足

class Solution {
    public int splitArray(int[] nums, int m) {
        int left = 0, right = 0;
        for (int i = 0; i < nums.length; i++) {
            right += nums[i];
            if (left < nums[i]) {
                left = nums[i];
            }
        }
        // left=当前最大值 right=数组和
        // 最大值在[单个最大,数组和]之间
        while (left < right) {
            int mid = (right - left) / 2 + left;
            if (check(nums, mid, m)) {
                // 分段数在m内，上界缩小，因为可以取所以直接取mid即可
                right = mid;
            } else {
                // 最大值mid太小，需要增加
                left = mid + 1;
            }
        }
        return left;
    }
    // 输入参数：数组nums，各自和的最大值x，段数m
    // 参数： sum当前分割子数组的和 cnt表示已分割的子数组数量（包括当前已分割的子数组数量）
    public boolean check(int[] nums, int x, int m) {
        int sum = 0;
        int cnt = 1;
        for (int i = 0; i < nums.length; i++) {
            if (sum + nums[i] > x) {
                // 当前段和超过最大，此时该分段
                cnt++;
                // sum进下一段
                sum = nums[i];
            } else {
                // 否则直接加nums[i];
                sum += nums[i];
            }
        }
        // 当前分段数在m之内
        return cnt <= m;
    }
}