cddiv/数组维护

时间:2023-03-09 03:54:53
cddiv/数组维护

题目连接

看代码:

#include <set>
#include <map>
#include <cmath>
#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
typedef long long LL;
using namespace std;
//const int inf=INT_MAX;
const int maxn=100000+100;
//#include <bits/stdc++.h>
int a[maxn];
int p1[maxn];
int main ()
{
int n;
while(~scanf("%d",&n))
{
memset(p1,0,sizeof(p1));
int sum=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
sum+=a[i];
}
if(sum%3)
printf("0\n");
else
{
LL ans=0;
int key=sum/3;
int sum1=0;
int sum2=0;
for(int i=n;i>=1;i--)
{
sum2+=a[i];
if(sum2==key)
p1[i-1]=p1[i]+1;
else
p1[i-1]=p1[i];
}
for(int i=1;i<=n;i++)
{
sum1+=a[i];
if(sum1==key)
{
ans+=p1[i+1];
}
}
printf("%lld\n",ans);
}
}
return 0;
}