leetcode刷题记录32-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 是 平衡数组 的 方案数 。

示例

示例 1：

输入：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 。不平衡。
只有一种让剩余数组成为平衡数组的方案。

示例 2：

输入：nums = [1,1,1]
输出：3
解释：你可以删除任意元素，剩余数组都是平衡数组。

示例 3：

输入：nums = [1,2,3]
输出：0
解释：不管删除哪个元素，剩下数组都不是平衡数组。

提示：

• 1 <= nums.length <= 10^5
• 1 <= nums[i] <= 10^4

问题分析：

这个题的难点就在于，随着元素的删除，该元素后续的奇偶节点发生变化。后续本来该是奇节点的，下标都会减一变成偶节点，偶节点同理。不管什么题，我都想暴力试一试万一过了呢。。。结果爆内存了。后来用了左右前缀和，最后是做出来了，我们首先计算原始数组的奇数偶数前缀和。

然后当删除元素时，我们很容易知道，该元素后续的原本为偶数和的变为了奇数和。假设我们用prefixEven来代表偶数前缀和，prefixOdd来代表奇数前缀和，假设我们要删除的元素下标为i，很容易知道，该下标前面的前缀和都不变，我们姑且定义为leftEven和leftOdd。那我们需要思考，rightEven是什么，很明显，原本的rightOdd,变成了rightEven，所以rightEven = rightOdd[n] - rightOdd[i + 1];详见代码。

代码如下：

class Solution {
public:
    int waysToMakeFair(vector<int>& nums) {
        int n = nums.size();
        //    奇偶前缀和
        vector<int> prefixEven(n + 1, 0);
        vector<int> prefixOdd(n + 1, 0);

        for (int i = 0; i < n; ++i) {
            prefixEven[i + 1] = prefixEven[i];
            prefixOdd[i + 1] = prefixOdd[i];
            if (i % 2 == 0) {
                prefixEven[i + 1] += nums[i];
            } else {
                prefixOdd[i + 1] += nums[i];
            }
        }
        int count = 0;

        for (int i = 0; i < n; ++i) {
            int leftEven = prefixEven[i];
            int leftOdd = prefixOdd[i];
            //    要删除掉i位置的元素
            int rightEven = prefixOdd[n] - prefixOdd[i + 1];
            int rightOdd = prefixEven[n] - prefixEven[i + 1];

            if (leftEven + rightEven == leftOdd + rightOdd) {
                count++;
            }
        }
        return count;
    }
};