Leetcode刷题笔记 410. 分割数组的最大值

题目
给定一个非负整数数组和一个整数 m，你需要将这个数组分成 m 个非空的连续子数组。设计一个算法使得这 m 个子数组各自和的最大值最小。

注意:
数组长度 n 满足以下条件:

1 ≤ n ≤ 1000
1 ≤ m ≤ min(50, n)

示例1
输入：
nums = [7,2,5,10,8]
m = 2
输出：
18

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

解法
动态规划（将数组分割为m段）

1. 注意如何创建数组，m段肯定是要拆开的，几个字符拆几个字符，字符数也要拆开
2. f[i][j]表示前i个数分成j段，所能得到的最大子数组和的最小值
3. 转移方程：
   1. 前i个数可以遍历，其中前k个[0:K]分为j-1段，k+1到i[k+1:i]为一段
   2. f[i][j] = min(max(f[k][j-1],nums_sum[k+1,i])
4. 注意边界
   1. 空字符不能为1段，所以i>=j
   2. f[0][0] = 0

代码

#include <stdio.h>
#include <vector>
#include <iostream>
using namespace std;
class Solution {
public:
    int splitArray(vector<int>& nums, int m) {
        int n = nums.size();
       vector<vector<long long>> f(n + 1, vector<long long>(m + 1, LLONG_MAX));
        vector<long long> sub(n + 1, 0);
        //计算前缀和，方便后面直接取，不需要循环
        for (int i = 0; i < n; i++) {
            sub[i + 1] = sub[i] + nums[i];
        }
        f[0][0] = 0;
        for (int i = 1; i <= n; i++) {//数组遍历
            for (int j = 1; j <= min(i, m); j++) {//切段遍历
                for (int k = 0; k < i; k++) {
                    f[i][j] = min(f[i][j], max(f[k][j - 1], sub[i] - sub[k]));
                }
            }
        }
        return (int)f[n][m];
    }
};
int main()
{
    vector<int> nums{7,2,5,10,8};
    int m=2;
    Solution s;
    cout<<s.splitArray(nums, m);
    return 0;
}

今天也是爱zz的一天哦！