Number of Ways CodeForces - 466C 分成三段相等数字和

题目链接：https://codeforces.com/problemset/problem/466/C?csrf_token=2d33d8e5911dbe22fd3b542450854920
转自：https://blog.csdn.net/SakuraKarma/article/details/104376785
题意：给一段序列，求能把序列分成三段连续相等的数字和的方法数。
思路：求一下前缀和，从1开始，枚举前缀和，统计下从1-（当前下标-1）的可以作为分段点的个数cnt，当以当前下标为分段点时，共有1 * cnt个选择，因此 当前下标对答案的贡献就是1*cnt。

#include <bits/stdc++.h>
using namespace std;
const int maxn=5e5+5;
long long a[maxn],pre[maxn],allnum;
int main()
{
    int n;
    allnum=0;
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%lld",&a[i]);
        pre[i]=pre[i-1]+a[i];
        allnum+=a[i];
    }
    if(allnum%3!=0)
    {
        printf("0");
        return 0;
    }
    long long ans=0,cnt=0;
    for(int i=1;i<n;i++)
    {
        if(pre[i]*3==allnum*2)
        {
            ans+=cnt;
        }
        if(pre[i]*3==allnum)
            cnt++;
    }
    printf("%lld\n",ans);
    return 0;
}