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

/* 
题目思路：
1. 使用Java Stream对数组进行处理。
2. 需要对流的元素进行聚合计算。
3. 使用Stream API来实现Kadane算法。
4. 需要考虑数组和前缀和、后缀和以及数组整体和的处理。
*/

import java.util.Arrays;

public class Solution {
    public int kConcatenationMaxSum(int[] arr, int k) {
        int MOD = 1000000007;
        long maxSubarraySum = kadane(arr);
        if (k == 1) {
            return (int)(maxSubarraySum % MOD);
        }
        long prefixSum = 0, suffixSum = 0, maxPrefixSum = 0, maxSuffixSum = 0;
        long totalSum = Arrays.stream(arr).asLongStream().sum();
        long currentSum = 0;
        for (int x : arr) {
            currentSum += x;
            prefixSum = Math.max(prefixSum, currentSum);
        }
        currentSum = 0;
        for (int i = arr.length - 1; i >= 0; i--) {
            currentSum += arr[i];
            suffixSum = Math.max(suffixSum, currentSum);
        }
        if (totalSum > 0) {
            return (int)((maxPrefixSum + maxSuffixSum + (totalSum * (k - 2))) % MOD);
        } else {
            return (int)(Math.max(maxPrefixSum + maxSuffixSum, maxSubarraySum) % MOD);
        }
    }
    private long kadane(int[] arr) {
        return Arrays.stream(arr)
            .mapToLong(x -> x)
            .collect(() -> new long[2], (a, b) -> {
                a[0] = Math.max(0, a[0] + b);
                a[1] = Math.max(a[1], a[0]);
            }, (a, b) -> {
                a[0] = Math.max(a[0], b[0]);
                a[1] = Math.max(a[1], b[1]);
            })[1];
    }
}