给定一个长度为 n 的数组 a1,a2,…,an。
现在，要将该数组从中间截断，得到三个非空子数组。
要求，三个子数组内各元素之和都相等。
请问，共有多少种不同的截断方法？
输入格式
第一行包含整数 n。
第二行包含 n 个整数 a1,a2,…,an。
输出格式
输出一个整数，表示截断方法数量。
数据范围
前六个测试点满足 1≤n≤10。
所有测试点满足 1≤n≤10^5，−10000≤ai≤10000。
扫描二维码关注公众号，回复：
14839226 查看本文章
输入样例1：
4
1 2 3 3
输出样例1：
1
输入样例2：
5
1 2 3 4 5
输出样例2：
0
输入样例3：
2
0 0
输出样例3：
0其实核心就是找两个点，一次枚举即可，前提是（sum % 3 == 0）第一次找一个和为sum / 3的点，第二次找一个和为sum / 3 * 2的点
AC代码如下
#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 ++ )
{
int x;
scanf("%d", &x);
s[i] = s[i - 1] + x;
}
if (s[n] % 3) puts("0");
else
{
LL res = 0, cnt = 0;
for (int j = 2; j < n; j ++ )
{
if (s[j - 1] == s[n] / 3) cnt ++ ;
if (s[j] == s[n] / 3 * 2) res += cnt;//状态传递方程
}
printf("%lld\n", res);
}
return 0;
}