题解 | 喜欢切数组的红

//***只要你 目光是瞄准月亮 迷失过 又有何妨***//
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
int n;
int a[N];
int pre[N];
int res[N];
bool flag;
int ans;
void solve()
{
    cin>>n;
    flag=0;
    ans=0;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    for (int i=1;i<=n;i++)
    {
        pre[i]=pre[i-1]+a[i];
    }
    if(pre[n]%3)
    {
        cout<<0<<'\n';
        return ;
    }
    for(int i=1;i<=n;i++)
    {
        res[i]=res[i-1]+(a[i]>0);
    }
    int x=pre[n]/3;
    for(int i=1;i<n-1;i++)
    {
        if((pre[i]!=x)||(res[i]==0))
        {
            continue;
        }
        for(int j=i+1;j<n;j++)
        {
            if((res[j]-res[i]==0)||((pre[j]-pre[i])!=x)||(res[n]-res[j]==0))continue;
            else ans++;

        }
    }
    cout<<ans<<'\n';
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int t;
    //cin>>t;
    t=1;
    while (t--)
    {
        solve();
    }
    return 0;
}