public int waysToMakeFair(int[] nums) {
int n = nums.length - 1;
if (n == 0) return 1;
int[] presum1 = new int[nums.length+1];//偶数
int[] presum2 = new int[nums.length+1];//奇数
presum1[0] = nums[0];
presum1[1] = nums[0];
presum2[1] = nums[1];
presum2[2] = nums[1];
for (int i = 2; i < nums.length; i+=2) {
presum1[i] = presum1[i-2] + nums[i];//求偶数前缀和
presum1[i+1] = presum1[i];
}
for (int i = 3; i < nums.length; i+=2) {
presum2[i] = presum2[i-2] + nums[i];//求奇数前缀和
presum2[i+1] = presum2[i];
}
int ans = 0;
if (presum1[n] - nums[0] == presum2[n]) ans++;//当删除第一个元素时
for (int i = 1; i < nums.length; i++) {//当删除后面元素时
if (i % 2 == 0) {//若删除的是偶数
//sum1为删除的数的前面求偶数和,后面求奇数和,其实就是求删除后的偶数和,下面同理
//第一个括号表示前面的偶数和,第二个括号表示后面的奇数和
int sum1 = (presum1[i] - nums[i]) + (presum2[n] - presum2[i-1]);
//sum2为删除的数前面求奇数和,后面求偶数和,其实就是求删除后的奇数和,下面同理
//第一个括号表示前面的奇数和,第二个括号遍是后面的偶数和
int sum2 = (presum2[i-1]) + (presum1[n] - presum1[i]);
if (sum1 == sum2) ans++;
}else {
int sum1 = (presum1[i-1]) + (presum2[n] - presum2[i]);
int sum2 = (presum2[i] - nums[i]) + (presum1[n] - presum1[i-1]);
if (sum1 == sum2) ans++;
}
}
return ans;
}
2023.1.28-----leetcode.1664
最新推荐文章于 2024-08-22 21:24:01 发布