CodeForces - 21C Stripe 2【思维题】

题目链接：https://codeforces.com/contest/21/problem/C

#include <iostream>
using namespace std;
static const int MAXN=1e5+10;
int a[MAXN];
int n,k;
long long sum,tmp,ans;
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        sum+=a[i];
    }
    if(sum%3) puts("0");
    else
    {
        sum/=3;
        for(int i=0;i<n-1;i++)
        {
            tmp=tmp+a[i];
            if(tmp==2*sum) ans=ans+k;
            if(tmp==sum) k++;
        }
        printf("%lld\n",ans);
    }
    return 0;
}