环形子数组的最大和

给定一个由整数数组 A 表示的环形数组 C,求 C 的非空子数组的最大可能和。

在此处,环形数组意味着数组的末端将会与开头相连呈环状。(形式上,当0 <= i < A.length 时 C[i] = A[i],且当 i >= 0 时 C[i+A.length] = C[i])

此外,子数组最多只能包含固定缓冲区 A 中的每个元素一次。(形式上,对于子数组 C[i], C[i+1], ..., C[j],不存在 i <= k1, k2 <= j 其中 k1 % A.length = k2 % A.length)

 

示例 1:

输入:[1,-2,3,-2]
输出:3
解释:从子数组 [3] 得到最大和 3
示例 2:

输入:[5,-3,5]
输出:10
解释:从子数组 [5,5] 得到最大和 5 + 5 = 10
示例 3:

输入:[3,-1,2,-1]
输出:4
解释:从子数组 [2,-1,3] 得到最大和 2 + (-1) + 3 = 4
示例 4:

输入:[3,-2,2,-3]
输出:3
解释:从子数组 [3] 和 [3,-2,2] 都可以得到最大和 3
示例 5:

输入:[-2,-3,-1]
输出:-1
解释:从子数组 [-1] 得到最大和 -1

 

#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

class Solution {
public:
    int maxSubarraySumCircular(vector<int> &A) {
        int maxPre = A.front(), minPre = A.front(), total = A.front(), maxX = A.front(), minX = A.front();
        for (int i = 1; i < A.size(); ++i) {
            maxPre = max(maxPre + A[i], A[i]);
            minPre = min(minPre + A[i], A[i]);
            maxX = max(maxX, maxPre);
            minX = min(minX, minPre);
            total += A[i];
        }
        if (total == minX)
            return maxX;
        return max((total - minX), maxX);
    }
};

int main() {
    vector<int> nums{-2, -3, -1};
    Solution s;
    cout << s.maxSubarraySumCircular(nums) << endl;
}

 

posted on 2021-03-04 16:14  QzZq  阅读(192)  评论(0)    收藏  举报

导航