【动态规划-最大子段和】力扣1191. K 次串联后最大子数组之和

给定一个整数数组 arr 和一个整数 k ,通过重复 k 次来修改数组。

例如,如果 arr = [1, 2] , k = 3 ,那么修改后的数组将是 [1, 2, 1, 2, 1, 2] 。

返回修改后的数组中的最大的子数组之和。注意,子数组长度可以是 0,在这种情况下它的总和也是 0。

由于 结果可能会很大,需要返回的 109 + 7 的 模 。

示例 1:

输入:arr = [1,2], k = 3

输出:9

示例 2:

输入:arr = [1,-2,1], k = 5

输出:2

示例 3:

输入:arr = [-1,-2], k = 7

输出:0

cpp 复制代码
class Solution {
public:
    int kConcatenationMaxSum(vector<int>& arr, int k) {
        int MOD = 1e9 + 7;
        long long sum = 0;
        long long maxAns = 0;
        long long pre = 0, maxMid = 0;
        for(int &num : arr){
            sum += num;
            maxAns = max(maxAns, sum);
            pre = max(pre + num, (long long)num);
            maxMid = max(maxMid, pre);
        }

        if(k == 1){
            return maxMid % MOD;
        }

        //最大后缀和
        long long maxAns2 = 0;
        long long sum2 = 0;
        for(int i = arr.size() - 1; i >= 0 ; i--){
            sum2 += arr[i];
            maxAns2 = max(maxAns2, sum2);
        }

        int theMax;
        if(sum > 0){
                theMax = max((k-2) * sum + maxAns + maxAns2, maxMid)%MOD;
        }
        else if(sum <= 0){
                theMax = max(maxMid,maxAns + maxAns2)%MOD;
        }
        return theMax;
    }
};

本题没有官解,也没有看到统一的比较好的题解。这道题主要是要考虑各种情况,可以先列出所有的情况,最后再进行代码优化,将类似的情况整理在一起。maxMid是最大子段和,使用的是Kadane算法,maxAns是最大前缀和,maxAns2是最大后缀和。首先如果当k为1的时候,最大子段和就是maxMid。接下来讨论sum的情况,如果sum>0,就要比较两种情况,一种是复制的2到k-1组元素的和加上第一组元素的最大后缀和再加上最后一组元素的最大前缀和,第二种情况是第一组元素中的最大子段和就是整个复制过的数组的最大子段和。当sum<=0的时候,也要比较两种情况哪个大,第一种是第一组元素中的最大子段和,第二种是最大前缀和加上最大后缀和(例如第一组元素的最大后缀和加上第二组元素的最大前缀和)。

相关推荐
算AI几秒前
LLM用于科学假设生成:探索与挑战
人工智能·算法
1白天的黑夜15 分钟前
贪心算法-2208.将数组和减半的最小操作数-力扣(LeetCode)
c++·算法·leetcode·贪心算法
格格Code15 分钟前
八大排序——冒泡排序/归并排序
数据结构·算法·排序算法
Dream it possible!28 分钟前
LeetCode 热题 100_最小路径和(92_64_中等_C++)(多维动态规划)
c++·leetcode·动态规划
vim怎么退出33 分钟前
46.二叉树展开为链表
前端·leetcode
C语言魔术师1 小时前
62.不同路径
算法·leetcode·动态规划
fantasy_41 小时前
LeetCode238☞除自身以外数组的乘积
java·数据结构·python·算法·leetcode
椰羊~王小美1 小时前
贪心算法和动态规划
算法·贪心算法·动态规划
圣保罗的大教堂1 小时前
leetcode 2302. 统计得分小于 K 的子数组数目 困难
leetcode
愚润求学1 小时前
【专题四】前缀和(3)
开发语言·c++·笔记·leetcode·刷题·c++11