题目链接:Codeforces 466C Number of Ways
题意:将给定序列分成和相等的三段,问有几种分法.
思路:依次判断前缀和,如果前缀和是总和的2/3时,将之前计算的前缀和是总和1/3的点累加即可.
注意:1.需要判断总和sum[n]是否可以整除3,否则会WA.
2.ans需为long long,否则精度不够.
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=5e5+5;
int a[maxn];
long long sum[maxn];
long long z1;
long long z2;
int main()
{
ios::sync_with_stdio(false);
sum[0]=0;
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
sum[i]=sum[i-1]+a[i];
}
z1=sum[n]/3;
z2=sum[n]/3*2;
long long ans=0;
if(sum[n]%3!=0)
{
cout<<0<<endl;
}
else
{
int cnt=0;
for(int i=1;i<n;i++)
{
if(sum[i]==z2)
{
ans+=cnt;
}
if(sum[i]==z1)
{
cnt++;
}
}
cout<<ans<<endl;
}
return 0;
}