AcWing 3956. 截断数组

给定一个长度为 n 的数组 a1,a2,…,an。

现在，要将该数组从中间截断，得到三个非空子数组。

要求，三个子数组内各元素之和都相等。

请问，共有多少种不同的截断方法？

输入格式

第一行包含整数 n。

第二行包含 n 个整数 a1,a2,…,an。

输出格式

输出一个整数，表示截断方法数量。

数据范围

前六个测试点满足 1≤n≤10。

所有测试点满足 1≤n≤105，−10000≤ai≤10000。

输入样例1：

4
1 2 3 3

输出样例1：

1

输入样例2：

5
1 2 3 4 5

输出样例2：

0

输入样例3：

2
0 0

输出样例3：

0

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=100010;
int n;
int s[N];
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&s[i]);
        s[i]+=s[i-1];
    }
    if(s[n]%3){
        puts("0");
        return 0;
    }
    LL res=0;
    for(int i=3,cnt=0;i<=n;i++){
        if(s[i-2]==s[n]/3) cnt++;
        if(s[n]-s[i-1]==s[n]/3) res+=cnt;
    }
    printf("%lld\n",res);
    return 0;
}

/* 
//通过15/22数据 
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int N=100010;
int s[N],a[N];
int main(){
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        if(i!=n) scanf(" ");
        s[i]=s[i-1]+a[i];
    } 
    if(n<3||s[n]%3!=0){
        printf("0\n");
        return 0;    
    } 
    if(s[n]==0){
        int ans=0;
        for(int i=1;i<=n;i++){
            if(s[i]==0) ans++;    
        }
        printf("%d\n",(ans-1)*(ans-2)/2);
        return 0;
    } 
    int flag=0,a,b;
    for(int i=0;i<n;i++){
        if(s[i]==s[n]/3){
            flag++;
            a=i;
        } 
        if(s[i]==s[n]/3*2){
            flag++;
            b=i;
        }    
    }
    if(flag==2&&b>a){
        printf("1");
        return 0;
    }
    else{
        printf("0"); 
        return 0;
    }
} 
*/