Codeforces Round #540 (Div. 3):B. Tanya and Candies

题目链接：http://codeforces.com/contest/1118/problem/B

题意：现在你有一个数列，你可以选择一个数去掉，去掉之后形成的新数列中奇数项和等于偶数项和，问去掉的这样的数有几个。

解题心得：去掉一个数之后这个数后面的数，奇数项和偶数项互换了。可以先得到整个数列的奇数项前缀和和偶数项前缀和，然后枚举每一个数，只要满足：前面奇数项和+后面偶数项和 == 前面偶数项和+后面奇数项和，就符合条件。

#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 2e5+100;

int num[maxn], n;
ll sum[3][maxn];//sum[0]代表偶数项的前缀和，sum[1]代表奇数项的前缀和

void init() {
    scanf("%d",&n);
    for(int i=1;i<=n;i++) {
        scanf("%d", &num[i]);
        if(i == 1) {
            sum[1][1] = num[i];
            continue;
        }
        sum[i%2][i] = sum[i%2][i-2] + num[i];
    }
}

int main()
{
    //freopen("1.in", "r", stdin);
    init();
    
    int ans = 0;
    for(int i=1;i<=n;i++) {
        ll pre1, pre2, last1, last2;//分别表示前奇数项和、前偶数项和，后奇数项和，后偶数项和
        pre1 = sum[i%2][i] - num[i];
        pre2 = sum[!(i%2)][i-1];
        if(n%2 == (!(i%2))) {
            last2 = sum[n%2][n] - sum[n%2][i-1];
            last1 = sum[i%2][n-1] - sum[i%2][i];
        } else {
            last2 = sum[!(i % 2)][n - 1] - sum[!(i % 2)][i - 1];
            last1 = sum[i%2][n] - sum[i%2][i];
        }

        if(pre1 + last2 == pre2 + last1) ans++;
    }
    printf("%d\n", ans);
    return 0;
}