LC410. 分割数组的最大值【贪心 + 二分答案】

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)

思路

这道题的经典解法是**"二分查找 + 贪心算法"**。遇到"求最大值的最小值"这类字眼,通常都可以优先考虑二分法。

核心解题思路:

  1. 锁定答案范围(二分边界): 无论怎么分割,子数组和的最大值,最小不会低于数组中的最大元素 (相当于切成最多份,k = nums.length),最大不会超过数组的元素总和 (相当于切成1份,k = 1)。这就是二分查找的上下界。
  2. 贪心验证(Check 函数): 假设当前我们猜想的最大和是 mid,我们如何验证它是否合法? 我们从左到右遍历数组,依次累加元素。如果加上当前元素后超过了 mid,说明当前子数组必须在这里截断,开启下一个子数组。遍历结束后,统计切出的子数组数量 count
  3. 调整二分方向:
    • 如果 count > k:说明切的份数太多了,意味着我们设定的 mid 太小,需要增大下界(left = mid + 1)。
    • 如果 count <= k:说明切的份数刚好或者还可以更少,意味着 mid 是一个合法的解,但可能还有更小的解,所以缩小上界继续逼近(right = mid)。
cpp 复制代码
// 注意到题目中的最小值最大 -> 二分答案😀
class Solution {
public:
    int splitArray(vector<int>& nums, int k) {
        int l = 0, r = 0;
        for(auto &x : nums) {
            r += x;
            l = max(l, x);
        }

        cout << "l, r: " << l << ' ' << r << endl;

        function<bool(int)> check = [&](int mid) {
            int cnt = 1, cur = 0;
            for(auto &x : nums) {
                if(cur + x > mid) ++ cnt, cur = x;
                else cur += x;
            }
            cout << "mid: " << mid << ' ' << "cnt: " << cnt << endl;
            return cnt <= k;
        };

        while(l < r) {
            int mid = l + r >> 1;
            if(check(mid)) r = mid;
            else l = mid + 1;
        }

        return l;
    }
};
相关推荐
王老师青少年编程11 小时前
csp信奥赛C++高频考点专项训练之贪心算法 --【反悔贪心】:建筑抢修
c++·算法·贪心·反悔贪心·csp·信奥赛·建筑抢修
王老师青少年编程12 小时前
csp信奥赛C++高频考点专项训练之贪心算法 --【反悔贪心】:种树
c++·算法·贪心·反悔贪心·csp·信奥赛·种树
王老师青少年编程2 天前
csp信奥赛C++高频考点专项训练之贪心算法 --【跳跃与过河问题】:过河问题
c++·算法·贪心·csp·信奥赛·跳跃与过河问题·过河问题
cpp_25012 天前
P1873 [COCI 2011/2012 #5] EKO / 砍树
数据结构·c++·算法·题解·二分答案·洛谷·csp
王老师青少年编程2 天前
csp信奥赛C++高频考点专项训练之贪心算法 --【跳跃与过河问题】:跳跳!
c++·算法·贪心·csp·信奥赛·跳跃与过河问题·跳跳
王老师青少年编程2 天前
csp信奥赛C++高频考点专项训练之贪心算法 --【跳跃与过河问题】:独木桥
c++·算法·贪心·csp·信奥赛·跳跃与过河问题·独木桥
cpp_25013 天前
P1024 [NOIP 2001 提高组] 一元三次方程求解
数据结构·c++·算法·题解·二分答案·洛谷·csp
王老师青少年编程4 天前
csp信奥赛C++高频考点专项训练之贪心算法 --【部分背包问题】:部分背包问题
c++·算法·贪心·csp·信奥赛·部分背包问题
王老师青少年编程4 天前
csp信奥赛C++高频考点专项训练之贪心算法 --【区间贪心】:种树
c++·算法·贪心·csp·信奥赛·区间贪心·种树
王老师青少年编程5 天前
csp信奥赛C++高频考点专项训练之贪心算法 --【区间贪心】:线段覆盖
c++·算法·贪心·csp·信奥赛·区间贪心·线段覆盖