LeetCode 410分割数组的最大值-二分+贪心

LeetCode 410 分割数组的最大值

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

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

示例 1：

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

示例 2：

输入：nums = [1,2,3,4,5], k = 2
输出：9

示例 3：

输入：nums = [1,4,4], k = 3
输出：4

提示：

• 1 <= nums.length <= 1000
• 0 <= nums[i] <= 106
• 1 <= k <= min(50, nums.length)

思路： 题目是一道二分题，给出一个数组nums和k（表示nums要分割的子序列个数）。要求在nums中划分k个子序列，每个子序列的求和值在各自序列和中最大，且在所有情况中该序列的和最小。是画匠问题的相似题，nums数组中每个元素表示当前需要画多久，k表示有多少个画匠，画匠只可以选择连续的序列进行画作，给出最短时间将nums画完

二分题步骤

1. 寻找答案范围，左区间端点为nums中的最大值，右区间的端点为只有一个画家时nums数组的求和值，[max, sum]
2. 确定该区间的单调性，可以看出二分区间为单调递增的
3. 确定f()函数逻辑，用贪心算法每个画家画作时长达到最大，比如需要每个画家画作10个小时，那么每个画家的画作时间都需要尽可能地接近10小时，当满足画匠分配完毕，且没有超出当前最优时间时退出，继续二分搜索最优时间。

代码

class Solution {
    public int splitArray(int[] nums, int k) {
      	// l为左端点 r为右端点 这里初始化为了避免答案出现在右端点位置
        int l = 0, r = 10;
        for (int i = 0; i < nums.length; i++) {
          	// 右端点为 nums的求和值，当且当只有一位画匠时所需要的最优时间
            r += nums[i];
          	// 左端点为 nums 中的最大耗时
            l = Math.max(l, nums[i]);
        }
      	// 只有一位画匠时 就返回最右端点即可
        if (k == 1) {
            return r;
        }
      	// 最优耗时
        int ans = 0;	
      	// 二分查找最优耗时
        while (l <= r) {
            int mid = (l + r) / 2;
          	// 当前耗时可行时 向左进行二分
            if (check(mid, nums, k)) {
                ans = mid;
                r = mid - 1;
            } else {
              	// 耗时不可行说明需要更高的耗时，所以需要向右进行二分
                l = mid + 1;
            }
        }
      	// 返回最优耗时即可
        return ans;
    }

    private static boolean check(int mid, int[] draws, int n) {
      	// 这里用到贪心算法 尽可能地让每个画匠的画作时间接近mid
        int tn = 0, count = 0, i = 0;
        for (; i < draws.length; i++) {
            if (count + draws[i] > mid) {
                tn++;
                count = draws[i];
            } else {
                count += draws[i];
            }
        }
      	// 返回条件即可
        return count <= mid && tn < n;
    }
}