经过一次操作后的最大子数组和

/*
 * Leetcode 1746: Maximum Subarray Sum After One Operation
 * 
 * 思路：
 * 1. 使用流和并行流结合进行数组遍历。
 * 2. 由于涉及到状态转移，不建议在流操作中使用副作用，采用传统for循环实现。
 */

import java.util.stream.IntStream;

public class Solution {
    public int maxSumAfterOperation(int[] nums) {
        int n = nums.length;
        int[] dp1 = new int[n];
        int[] dp2 = new int[n];
        dp1[0] = nums[0];
        dp2[0] = nums[0] * nums[0];
        int maxSum = Math.max(dp1[0], dp2[0]);

        IntStream.range(1, n).forEach(i -> {
            dp1[i] = Math.max(dp1[i - 1] + nums[i], nums[i]);
            dp2[i] = Math.max(dp1[i - 1] + nums[i] * nums[i], Math.max(nums[i] * nums[i], dp2[i - 1] + nums[i]));
            maxSum = Math.max(maxSum, Math.max(dp1[i], dp2[i]));
        });

        return maxSum;
    }
}