LeetCode 1191 K次串联后最大子数组之和

题目描述：
给你一个整数数组 arr 和一个整数 k。

首先，我们要对该数组进行修改，即把原数组 arr 重复 k 次。

举个例子，如果 arr = [1, 2] 且 k = 3，那么修改后的数组就是 [1, 2, 1, 2, 1, 2]。

然后，请你返回修改后的数组中的最大的子数组之和。

注意，子数组长度可以是 0，在这种情况下它的总和也是 0。

由于 结果可能会很大，所以需要 模（mod） 10^9 + 7 后再返回。

思路：
先计算两个周期的最大子数组之和
然后再加上其他周期的，与上个周期的比较

代码如下：

class Solution {
public:
    int kConcatenationMaxSum(vector<int>& arr, int k) {
        vector<int>temp=arr;
        long sum=0;
        int s=0;
        int max_s=0;
        int mod=1000000007;
        if(k<2){//对k==1进行处理
            for(int i=0;i<arr.size();i++){
               sum+=arr[i];
               s+=arr[i];
               if(s>0)
               max_s=max(s,max_s);
               else s=0;
            }
            return max_s;
        }
        else{
            temp.insert(temp.end(),arr.begin(),arr.end());
            for(int i=0;i<arr.size();i++){//计算两个周期的最大子数组
               sum=sum+arr[i];
            }
            for(int i=0;i<temp.size();i++){
               s+=temp[i];
               if(s>0)
               max_s=max(s,max_s);
               else s=0;
            }
        }
        if(k==2||sum<0) 
        return max_s;
        return (sum*(k-2)+max_s)%mod;//加上其他周期进行比较
    }
};