题意:给出一个序列,求出这个序列切成三段相同值的方案数
首先sum值一定整除3,否则方案数为0,接着预处理出sum/3和sum/3*2的下标,再扫一遍就能得到答案
#include<cstring>
#include<string>
#include<iostream>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<map>
#include<cstdlib>
#include<cmath>
#include<vector>
//#pragma comment(linker, "/STACK:1024000000,1024000000");
using namespace std;
#define INF 0x3f3f3f3f
#define maxn 500005
int n;
long long a[maxn];
long long pre[maxn];
int vis1[maxn],vis2[maxn];
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%I64d",&a[i]);
pre[i]=pre[i-1]+a[i];
}
long long sum=pre[n];
if(sum%3)
{
printf("0\n");
return 0;
}
for(int i=1;i<n;i++)
{
if(pre[i]==sum/3) vis1[i]=1;
if(pre[i]==sum/3*2) vis2[i]=1;
}
int num=0;
long long ans=0;
for(int i=1;i<=n;i++)
{
if(vis2[i]) ans+=num;
if(vis1[i]) num++;
}
printf("%I64d\n",ans);
return 0;
}