动态规划学习

三个无重叠子数组的最大和（leetcode 689）

给你一个整数数组 nums 和一个整数 k ，找出三个长度为 k 、互不重叠、且 3 * k 项的和最大的子数组，并返回这三个子数组。以下标的数组形式返回结果，数组中的每一项分别指示每个子数组的起始位置（下标从 0 开始）。如果有多个结果，返回字典序最小的一个。

示例 1：
输入：nums = [1,2,1,2,6,7,5,1], k = 2
输出：[0,3,5]
解释：子数组 [1, 2], [2, 6], [7, 5] 对应的起始下标为 [0, 3, 5]。
也可以取 [2, 1], 但是结果 [1, 3, 5] 在字典序上更大。

思路：前缀和加动态规划，由后向前回溯。

状态转移方程为： dp[i][j] = Math.max(dp[i-1][j],dp[i - k][j - 1] + sum[i] - sum[i - k]);

public int[] maxSumOfThreeSubarrays(int[] nums, int k) {
        int len = nums.length;
        reverse(nums);
        long []sum = new long[len + 1];
        long [][]dp = new long[len + 2][4];
        for(int i = 1; i <= len; i++)
            sum[i] = sum[i-1] + nums[i-1];
        for(int i = k; i <= len; i++){
            for(int j = 1; j < 4; j++){
                dp[i][j] = Math.max(dp[i-1][j],dp[i - k][j - 1] + sum[i] - sum[i - k]); 
            }
        }
        int []result = new int[3];
        int  j = 3, i = len,index = 0;
        while(j > 0){
            if(dp[i-1][j] > dp[i - k][j - 1] + sum[i] - sum[i - k]){
                i--;
            }else{
                result[index++] = len - i;
                i = i - k;
                j--;
            }
        }
        return result;
    }
    public void reverse(int []nums){
        int left = 0,right = nums.length - 1;
        while(left < right){
            int temp = nums[right];
            nums[right--] = nums[left];
            nums[left++] = temp;
        }
    }

如果从后向前回溯，字典序就是最大的，因为在从后向前回溯的过程中，每一步迭代需要调用max函数，由于max函数的特性，只要在靠后的位置获得了局部的max，前面的位置即便等于max也不会再更新，所以这样回溯得到的字典序是最大的，所以将数组翻转，这意味着我们先将原来数组的字典序颠倒了过来，颠倒后数组的最大字典序变成了原来数组的最小字典序，因此求出来的是原来数组的最小字典序。