Leetcode 689-三个无重叠子数组的最大和

题目描述

给定数组 nums 由正整数组成，找到三个互不重叠的子数组的最大和。

每个子数组的长度为k，我们要使这3*k个项的和最大化。

返回每个区间起始索引的列表（索引从 0 开始）。如果有多个结果，返回字典序最小的一个。

题解

用dp[i][j]代表到j元素为止，在有i个子数组的情况下最大的值。之后再回过头找最早得到最大和的位置。相比官解的方法来说应该更直白易懂些。

转移方程的意思是：对于dp[i][j]，其最大值是下面两种情况之一：

要么在j-1个元素的时候已经找到了i个数组的最大值；
要么j-k个元素为止有i-1个数组的最大值，然后加上结束于第j个元素的数组的和形成i个数组的最大值。

class Solution {

    public int[] maxSumOfThreeSubarrays(int[] nums, int k) {
        // 状态定义：dp[i][j]表示遍历到i时子数组数量为j时的最大值
        // 状态转移：dp[i][j]=max(dp[i-k][j-1]+sum(i), dp[i-1][j])
        // 初始值：dp[i][0]=0
        // 返回值：dp[n-k][3]

        int n = nums.length;
        int[] sums = new int[n+1];

        // 前缀和，用于后面计算区间和
        //sums[i]对应于nums[0]~nums[i-1]
        for (int i = 1; i <= n; i++) {
            sums[i] = sums[i - 1]+nums[i-1];
        }

        // 动态规划，注意下标越界的特殊处理
        int[][] dp = new int[n][4];
        for (int i = 0; i <= n - k; i++) {
            for (int j = 1; j <= 3; j++) {
                if (i == 0 ) {
                    //边界条件：i=0时
                    //dp[0][1]=前k个数的和，i为0时子数组最大值只能为前k个数的和
                    //dp[0][2],dp[0][3]不重要，因为以零开头不可能包含三/二个数组
                    dp[i][j] = sums[i+k]-sums[i];
                } else if (i < k) {
                    //i<k时dp[i][1]值固定，不可能由dp[i - k][j - 1]转移来
                    //sums[i+k]-sums[i]为nums[i]~nums[i+k-1]的和
                    dp[i][j] = Math.max(sums[i+k]-sums[i], dp[i - 1][j]);
                } else {
                    dp[i][j] = Math.max(dp[i - k][j - 1] + sums[i+k]-sums[i], dp[i - 1][j]);
                }
            }
        }

        // 处理结果
        int[] ans = new int[3];
        // 从后往前遍历
        int i = n - k, j = 3, idx = 2;
        while (j >= 1) {
            if (i == 0) {
                ans[0] = 0;
                break;
            } else if (dp[i][j] == dp[i - 1][j]) {
                i--;
            } else {
                ans[idx--] = i;
                i -= k;
                j--;
            }
        }

        return ans;
    }
}