- 生成平衡数组的方案数
给你一个整数数组 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 <= 105
1 <= nums[i] <= 104
通过次数2,644
提交次数5,331
题意:给定一个数组arr[n]
,你可以删除其中一个
位置i
(删除后i
往后的下标会往前移动),问有多少种方案使得
(
∑
i
=
1
且
i
是
奇
数
n
a
r
r
[
i
]
)
=
=
(
∑
i
=
1
且
i
是
偶
数
n
a
r
r
[
i
]
)
(\sum_{i=1且i是奇数}^{n}arr[i]) == (\sum_{i=1且i是偶数}^{n}arr[i])
(∑i=1且i是奇数narr[i])==(∑i=1且i是偶数narr[i])
1e5
个数,只能删除一个,可以O(n)
枚举删除每个位置后是否奇和 == 偶和
- 开两个数组
ji_sum[n]和ou_sum[n]
分别存储奇下标前缀和
和偶下标前缀和
- 删除
i
后,前缀[i+1, n]
会发生奇偶交换
#define MAXN ((int)1e5+7)
int sum_ji[MAXN], sum_ou[MAXN];
class Solution {
public:
int waysToMakeFair(vector<int>& vec) {
int n = vec.size();
for(int i=0; i<n; i++) {
int pos = i & 1;
sum_ji[i+1] = sum_ji[i] + (pos ? vec[i] : 0); //奇下标前缀和
sum_ou[i+1] = sum_ou[i] + (pos ? 0 : vec[i]); //偶下标前缀和
}
int ans = 0;
for(int i=1; i<=n; i++) {
int ji = sum_ji[i-1];
int ou = sum_ou[i-1];
int jrig = sum_ji[n] - sum_ji[i];
int orig = sum_ou[n] - sum_ou[i];
//删除i后前缀和[i+1, n]奇偶交换
if(ji+orig == ou+jrig) ans ++;
}
return ans;
}
};