【Leetcode】410. Split Array Largest Sum（配数学证明）

题目地址：

https://leetcode.com/problems/split-array-largest-sum/

给定一个正整数数组$A$，再给定一个正整数$m$，要求将$A$分为若干连续的子段，使得每个子段的和的最大值最小。返回那个最小和。

动态规划做法可以参考https://blog.csdn.net/qq_46105170/article/details/109114167。下面介绍二分答案的做法。

答案可能的区间是$[l,r]=[\max A,\sum A]$，对应着分$l_A$段和分$1$段。接着对于某个答案$x$进行判定。我们要找的是最小的$x$，使得$A$能分为$m$段，每段的和都小于等于$x$。判定方式是，先从$A[0]$开始，每次贪心地累加$A$的一段，直到恰好到最后一个小于等于$x$的位置为止，这一段就分出来了，然后再开始下一段的累加，直到整个$A$被分割完毕。接着开始判定，如果$A$被拆为小于等于$m$段，那显然也能拆为$m$段也使得每段的和小于等于$x$，说明$x$已经满足条件了，答案应该在$[l,x]$中；如果$A$被分为大于$m$段，那说明$x$设置的太小了，应该在$(x,r]$中寻找答案（这里需要证明一下一定不存在将$A$分为$m$段且每段和都小于等于$x$的分法。思路是数学归纳法和反证法。首先当$A$长度是$1$那显然成立。接着反证法，如果存在，不妨设这种分法的第一段的长度$s$是小于上面判定所使用的分法分出来的第一段长度$t$，这样就存在$A[s+1:l_A-1]$的一种分法可以分为$m-1$段使得每段和小于等于$x$，直接把分割点分在$A[t+1:l_A-1]$上，就得到了$A[t+1:l_A-1]$的$m-1$段的分法，但由归纳假设，贪心法得到的段数已经是最小的了，矛盾）。代码如下：

class Solution {
 public:
  int splitArray(vector<int>& A, int k) {
    int l = 0, r = 0;
    for (auto x : A) l = max(l, x), r += x;
    while (l < r) {
      int mid = l + (r - l >> 1);
      if (check(A, k, mid)) r = mid;
      else l = mid + 1;
    }

    return l;
  }

  bool check(vector<int>& A, int k, int sum) {
    int cnt = 1, ksum = 0;
    for (auto x : A) {
      if (ksum + x <= sum) ksum += x;
      else cnt++, ksum = x;

      if (cnt > k) return false;
    }

    return cnt <= k;
  }
};

时间复杂度$O(l_A\log \sum A_i)$，这里求和只对$A$把最大值删掉的部分求和，空间复杂度$O(1)$。