LC1664生成平衡数组的方案数

题目描述

给你一个整数数组 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 是 平衡数组 的 方案数

前缀和+后缀和

    private static int solve(int[] nums) {
        int n = nums.length;
        int ans = 0;
        int[] pre = new int[n + 1];     // 前缀和
        int[] suf = new int[n + 1];     // 后缀和

        for (int i = 1; i <= n; i++) {
            pre[i] = nums[i - 1];
            if (i - 2 >= 1) pre[i] += pre[i - 2];  // 奇偶性一致，相加
        }

        for (int i = n; i >= 1; i--) {
            suf[i] = nums[i - 1];
            if (i + 2 <= n) suf[i] += suf[i + 2];  // 奇偶性一致，相加
        }

        // 统计,i表示需要删除的第i个元素
        for (int i = 1; i <= n; i++) {
            int odd = 0;  // 与当前需要删除的第i个元素 奇偶性一致的
            int even = 0;   // 与当前需要删除的第i个元素 奇偶性相反的

            if (i - 2 >= 1) odd += pre[i - 2];
            if (i + 1 <= n) odd += suf[i + 1];

            if (i - 1 >= 1) even += pre[i - 1];
            if (i + 2 <= n) even += suf[i + 2];

            if (odd == even) ans++;
        }

        return ans;
    }