LeetCode 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 是 平衡数组 的 方案数 。

示例
输入：nums = [2,1,6,4]
输出：1
解释：
删除下标 0 ：[1,6,4] -> 偶数元素下标为：1 + 4 = 5 。奇数元素下标为：6 。不平衡。
删除下标 1 ：[2,6,4] -> 偶数元素下标为：2 + 4 = 6 。奇数元素下标为：6 。平衡。
删除下标 2 ：[2,1,4] -> 偶数元素下标为：2 + 4 = 6 。奇数元素下标为：1 。不平衡。
删除下标 3 ：[2,1,6] -> 偶数元素下标为：2 + 6 = 8 。奇数元素下标为：1 。不平衡。
只有一种让剩余数组成为平衡数组的方案。

解决办法:

一般的想法是从头开始遍历数组，对每个元素进行判断，看数组上删除这个元素之后是不是平衡数组，但是这种方法的时间复杂度比较高，有一种方法是将这个数组一删除的那个位置为界限，把数组划分为左右数组，对于 奇数/偶数 下标和来说，左边的 奇数/偶数 下标和是不变的，右边的奇数下标和与左边的偶数下标和对应，右边的偶数下标和与左边的奇数下标和对应，循环每次判断
左奇数下标和 + 右偶数下标和 == 右奇数下标和 + 左偶数下标和
如果相等，那么删除这个位置就是一种方案。
代码如下:

    public int waysToMakeFair(int[] nums) {
       int presum1 = 0, presum2 = 0;
       int sum1 = 0, sum2 = 0;  // sum1: 奇数和 , sum2: 偶数和
       for(int i=0; i<nums.length; i++){
            if(i%2==0)
                sum2 += nums[i];
            else
                sum1 += nums[i];
       }

       int res = 0;
       for(int i=0; i<nums.length; i++){
           if(i%2 == 0){
               sum2 -= nums[i];
               if(sum2 + presum1 == sum1 + presum2)
                    res++;
               presum2 += nums[i]; 
           }
           else{
               sum1 -= nums[i];
               if(sum2 + presum1 == sum1 + presum2)
                    res++;
               presum1 += nums[i]; 
           }
       }

       return res;
    }