Acwing蓝桥杯每日一题2.13截断数组

题目链接：3956. 截断数组 - AcWing题库

解体思路：一开始想要用双指针，但最坏时间复杂度是n2，显然超时

正确做法：要把一组数分成三等分，那么每等分的和avg是总和的三分之一

找到前缀和等于avg*2的地方（i），答案加上i之前前缀和为avg的个数

以此类推得到最终答案。

#include
using namespace std;
const int N = 1e5+10;
int a[N];
int main()
{
    int n;
    long long sum = 0;
    cin >> n;
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]), sum += a[i];
    if (sum % 3 != 0 || n <= 2)//满足这些条件直接return
    {
        printf("0");
        return 0;
    }
    
    long long avg = sum / 3;
    long long fs = 0, ans = 0, s = 0;
    for (int i = 1; i < n; i++)
    {
        s += a[i];
        if (s == avg * 2) ans += fs;
        if (s == avg) fs++;
    //此处注意两个if语句不能颠倒。若是例子0 0 0就错误了
    }
    printf("%lld", ans);
    return 0;
}