美文网首页算法算法提高之LeetCode刷题数据结构和算法分析
689-三个无重叠子数组的最大和-通过动态规划优化

689-三个无重叠子数组的最大和-通过动态规划优化

作者: 华雨欣 | 来源:发表于2020-04-05 13:46 被阅读0次

    题目

    核心思路

    这道题应该算是困难题目里比较简单的一道了,最关键的就是知道怎么优化来提高效率。首先给出一种暴力法,先穷举出所有的可能,然后去发现可以优化的地方。

    暴力法

    想要穷举所有可能,对于题目中提到的3个子数组,显然需要至少三层循环,我们设i , j , m分别表示三个子数组的结尾下标(开头也相同),那就需要三层循环来遍历i , j , m,然后还要从这三个下标分别向前求 k 个数的和,也就是说总共需要四层循环,复杂度十分可怕,代码就不给出了。

    优化一

    这是一种很常见的方法,就是求前缀和,通过前缀和来求子数组的和可以省去一层循环,即知道结尾下标之后向前遍历求和的过程。

    代码
    class Solution {
        public int[] maxSumOfThreeSubarrays(int[] nums, int k) {
            int[] ans = new int[3];
            int[] sum = new int[nums.length + 1];
            for (int i = 0; i < nums.length; i++) {
                //求前缀和
                sum[i + 1] = sum[i] + nums[i];
            }
            for (int i = nums.length; i >= k; i--) {
                //求长度为k的子数组的间隔和
                sum[i] = sum[i] - sum[i - k];
            }
            int maxSum = 0;
            //遍历
            for (int i = k; i <= nums.length - 2 * k; i++) {
                for (int j = i + k; j <= nums.length - k; j++) {
                    for (int m = j + k; m <= nums.length; m++) {
                        if (maxSum < sum[i] + sum[j] + sum[m]) {
                            maxSum = sum[i] + sum[j] + sum[m];
                            ans[0] = i - k;
                            ans[1] = j - k;
                            ans[2] = m - k;
                        }
                    }
                }
            }
            return ans;
        }
    }
    

    思路仍是暴力,但是通过求前缀和的预处理,把最内层的循环省掉,提高一些效率。

    优化二

    一种直观的想法:如果只找数组中一个最大的数,直接用一个变量遍历一遍即可;那么如果i , j , m的下标范围是确定的,问题就会变得同上边的想法,变得很简单。不过这道题要求的是长度为k的子数组,范围就是可变的了。但是如果固定中间位置的下标 j ,此时他前后两个子数组 i , m的下标就变成固定范围了,固定范围肯定就会有一个最大值(即最优解),问题有局部最优解,也就须要用到DP了
    所以我们使用一个 left 数组保存第一个子数组的局部最优解,用一个 right 数组保存最后一个子数组的局部最优解,然后遍历每一个可能中间下标 j,就可以求出最终答案,并且可以省去两层循环,效率大大提高。

    代码
    class Solution {
        public int[] maxSumOfThreeSubarrays(int[] nums, int k) {
            int[] ans = new int[3];
            int[] sum = new int[nums.length + 1];// 长度+1,sum[0] = 0方便求前缀和
            int[] left = new int[nums.length + 1];// 长度与sum一致方便运算、理解
            int[] right = new int[nums.length + 1];// 同上
            // 求前缀和
            for (int i = 0; i < nums.length; i++) {
                sum[i + 1] = sum[i] + nums[i];
            }
            // 求间隔为k的和,从下标k开始求,以结尾为主
            for (int i = nums.length; i >= k; i--) {
                sum[i] = sum[i] - sum[i - k];
            }
            // 求从左边开始,和最大的值最早出现的下标
            int maxNum = 0;
            int index = 0;
            for (int i = k; i <= nums.length - k * 2; i++) {
                if (maxNum >= sum[i]) {// >=号保证存储的是最先出现的下标
                    left[i] = index;
                } else {
                    left[i] = i;
                    index = i;
                    maxNum = sum[i];
                }
            }
            // 求从右边开始,和最大的值最早出现的下标
            maxNum = 0;
            index = 0;
            for (int i = nums.length; i >= k * 2; i--) {
                if (maxNum > sum[i]) {// >号保证存储的是最先出现的下标
                    right[i] = index;
                } else {
                    right[i] = i;
                    maxNum = sum[i];
                    index = i;
                }
            }
            // 遍历j可能的位置,求出所有可能的最大值
            maxNum = 0;
            for (int i = k * 2; i <= nums.length - k; i++) {
                if (maxNum < sum[i] + sum[left[i - k]] + sum[right[i + k]]) {
                    maxNum = sum[i] + sum[left[i - k]] + sum[right[i + k]];
                    ans[0] = left[i - k] - k;// - k由于求间隔和的时候,和是存储在每个间隔最后一个元素的下标
                    ans[1] = i - k;
                    ans[2] = right[i + k] - k;
                }
            }
    
            return ans;
        }
    }
    

    我这里将和数组、left数组、right数组大小都设为 nums.length + 1,主要是为了方便运算,也方便理解,实际上用到的大小只有 nums.length - k 罢了。如有内容错误的地方还请指出,感谢相遇~

    相关文章

      网友评论

        本文标题:689-三个无重叠子数组的最大和-通过动态规划优化

        本文链接:https://www.haomeiwen.com/subject/qoluphtx.html