分割数组的最多方案数

/*
 * 思路：
 * 1. 计算数组的总和totalSum。
 * 2. 使用Java Stream遍历数组，计算每个可能的pivot左侧和右侧的和。
 * 3. 如果找到符合条件的pivot，增加计数。
 * 4. 尝试将数组中的每个元素修改为k，再次检查是否符合条件。
 */
import java.util.stream.IntStream;

public class Solution {
    public int waysToPartition(int[] nums, int k) {
        int n = nums.length;
        long totalSum = IntStream.of(nums).asLongStream().sum();
        long[] leftSums = new long[n];
        long[] rightSums = new long[n];

        leftSums[0] = nums[0];
        for (int i = 1; i < n; i++) {
            leftSums[i] = leftSums[i - 1] + nums[i];
        }

        rightSums[n - 1] = nums[n - 1];
        for (int i = n - 2; i >= 0; i--) {
            rightSums[i] = rightSums[i + 1] + nums[i];
        }

        long ways = IntStream.range(1, n)
            .filter(pivot -> leftSums[pivot - 1] == rightSums[pivot])
            .count();

        for (int i = 0; i < n; i++) {
            long newTotalSum = totalSum - nums[i] + k;

            ways += IntStream.range(1, n)
                .filter(pivot -> (pivot - 1 == i ? k : leftSums[pivot - 1]) == (pivot == i ? k : rightSums[pivot]))
                .count();
        }

        return (int) ways;
    }
}