解体思路:一开始想要用双指针,但最坏时间复杂度是n2,显然超时
正确做法:要把一组数分成三等分,那么每等分的和avg是总和的三分之一
找到前缀和等于avg*2的地方(i),答案加上i之前前缀和为avg的个数
以此类推得到最终答案。
#include<iostream>
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;
}