leetcode.1664 生成平衡数组的方案数 - 奇偶性前缀和

1664. 生成平衡数组的方案数

题目：

给你一个整数数组 nums 。你需要选择 恰好 一个下标（下标从 0 开始）并删除对应的元素。请注意剩下元素的下标可能会因为删除操作而发生改变。

比方说，如果 nums = [6,1,7,4,1] ，那么：

• 选择删除下标 1 ，剩下的数组为 nums = [6,7,4,1] 。
• 选择删除下标 2 ，剩下的数组为 nums = [6,1,4,1] 。
• 选择删除下标 4 ，剩下的数组为 nums = [6,1,7,4] 。

如果一个数组满足奇数下标元素的和与偶数下标元素的和相等，该数组就是一个 平衡数组 。

请你返回删除操作后，剩下的数组 nums 是 平衡数组 的 方案数 。

思路：

我们可以枚举每一个被删除的坐标

被删除的坐标i前面的坐标位置不变，后面的坐标奇偶坐标位置交换

因此先分别求出nums数组的奇偶前缀和

如果【前奇数之和+后偶数之和==前偶数之和+后奇数之和】则res++

注：因为后面坐标位置奇偶互换

比如：原始数组 nums = [6,1,7,4,1] —— [0,1,2,3,4]

选择删除下标 1 ，剩下的数组为 nums = [6,7,4,1] —— [0,1,2,3]

可以看出被删元素1后面的元素坐标奇偶坐标交换

class Solution {
    static int N=100050;
    static int[] s1=new int[N],s2=new int[N]; //s1偶数前缀和 s2奇数前缀和
    public int waysToMakeFair(int[] nums) {
        int res=0,n=nums.length;
        int[] a=new int[n+1];
        for(int i=1;i<=n;i++) a[i]=nums[i-1];
        for(int i=1;i<=n;i++)
            if(i%2==0)
            {
                s1[i]=a[i]+s1[i-1]; 
                s2[i]=s2[i-1];
            } 
            else
            {
                s1[i]=s1[i-1];
                s2[i]=s2[i-1]+a[i];
            }
        for(int i=1;i<=n;i++)
        {
            int pre1=s1[i-1],pre2=s2[i-1];
            int next1=s1[n]-s1[i],next2=s2[n]-s2[i]; //i后面的奇偶顺序交换
            if(pre1+next2==pre2+next1) res++;
        }
        return res;
    }
}