分成3份,那么总和sum一定可以被3整除,然后找到一个等于总和2/3的地方,那么结果就是在前面段和为总和1/3的个数
代码
#include<iostream>
using namespace std;
int n;
const int N=1e6+10;
int q[N];
#define ll long long
ll sum=0;
int main()
{
cin>>n;
for(int i=0;i<n;i++)cin>>q[i],sum+=q[i];
if(sum%3==0)cout<<0<<endl;
else
{
ll as=sum/3;
ll tot=0,js=0;
ll ans=0;
for(int i=0;i<n;i++)
{
tot+=q[i];
if(tot==as*2)ans+=js;
if(tot==as)js++;
}
cout<<ans<<endl;
}
return 0;
}