生成平衡数组的方案数

// Java Stream Solution

/*
 * 思路：
 * 1. 计算初始数组中奇数和偶数下标的元素之和。
 * 2. 遍历数组，每次删除一个元素后，重新计算剩余数组的奇数和偶数下标元素之和，判断是否为平衡数组。
 * 3. 使用流的方式处理前缀和的计算，以优化计算过程。
 */

import java.util.stream.IntStream;

public class StreamSolution {
    public int waysToMakeFair(int[] nums) {
        int n = nums.length;
        int totalOddSum = IntStream.range(0, n).filter(i -> i % 2 != 0).map(i -> nums[i]).sum();
        int totalEvenSum = IntStream.range(0, n).filter(i -> i % 2 == 0).map(i -> nums[i]).sum();

        int leftOddSum = 0, leftEvenSum = 0;
        int count = 0;
        for (int i = 0; i < n; i++) {
            if (i % 2 == 0) {
                totalEvenSum -= nums[i];
            } else {
                totalOddSum -= nums[i];
            }

            if (leftOddSum + totalEvenSum == leftEvenSum + totalOddSum) {
                count++;
            }

            if (i % 2 == 0) {
                leftEvenSum += nums[i];
            } else {
                leftOddSum += nums[i];
            }
        }

        return count;
    }
}