题解 | 喜欢切数组的红

#include <iostream>
using namespace std;

long long v[200005],pre_sum[200005],aft_sum[200005];

int main() {
    long long n;
    cin>>n;
    long long num=0,sum=0;
    for(int i=1;i<=n;i++){
        cin>>v[i];
        sum+=v[i];
        if(v[i]>0)num++;
        pre_sum[i]=pre_sum[i-1]+v[i];
    }
    // cout<<sum<<'\n';
    if(sum%3!=0){
        cout<<0;
    }
    else{
        long long ans=0;
        sum/=3;
        // cout<<sum<<'\n';
        for(int i=n;i>=1;i--){
        aft_sum[i]=aft_sum[i+1]+v[i];
    }
    int flag1=0;
    for(int i=1;i<=n-2;i++){
        if(v[i]>0){
            num--;
            flag1=1;
        }
        long long num1=num;
        if(pre_sum[i]==sum&&flag1&&num>0){
            int flag2=0;
            for(int j=n;j>i;j--){
                if(v[j]>0){
                    num1--;
                    flag2=1;
                }
                if(aft_sum[j]==sum&&flag2&&num1>0){
                    ans++;
                }
            }
        }
    }
    cout<<ans;
    }
    
}
// 64 位输出请用 printf("%lld")