力扣1031. 两个非重叠子数组的最大和

力扣1031. 两个非重叠子数组的最大和

题目解析及思路

题目要求找到两段长分别为firstLen 和 secondLen的子数组，使两段元素和最大

图解见灵神

枚举第二段区间的右端点，在左边剩余部分中找出元素和最大的第一段区间，并用前缀和优化求子数组元素和

代码

class Solution {
public:
    int maxSumTwoNoOverlap(vector<int>& nums, int firstLen, int secondLen) {
        int ans = 0,n = nums.size(),s[n+1];
        s[0] = 0;
        //求前缀和
        for(int i=1;i<=n;i++)
            s[i] = s[i-1] + nums[i-1];
        auto f = [&](int firstLen,int secondLen)
        {
            int maxsum = 0;
            //枚举第二段区间右端点
            for(int i = firstLen+secondLen;i<=n;i++)
            {
                //求第一段区间最大值
                maxsum = max(maxsum,s[i-secondLen] - s[i-secondLen-firstLen]);
                ans = max(ans,maxsum + s[i] - s[i-secondLen]);
            }
        };
        //左a右b
        f(firstLen,secondLen);
        //左b右a
        f(secondLen,firstLen);
        return ans;
    }
};