【题】【前缀和】NKOJ 3049 三分数组

NKOJ 3049 三分数组

时间限制 : 10000 MS 空间限制 : 65536 KB

问题描述

给出一个有n 个整数的数组a[1],a[2],…,a
, 有多少种方法把数组分成3 个连续的子序列，使得各子序列的元素之和相等。也就是说，有多少个下标对i,j (2≤i≤j≤n-1), 满足：sum(a[1]..a[i-1]) = sum(a[i]..a[j]) = sum(a[j+1]..a
)

输入格式

第1 行：1 个整数n(1 <= n <= 5*10^5)

接下来n 行，每行1 个整数，表示a[i](|a[i]| <= 10^9)

输出格式

第1 行:1 个整数，表示答案，如果不能3 等分，输出0

样例输入

Sample1:

5

1 2 3 0 3

Sample2:

4

0 1 -1 0

Sample3:

2

4 1

样例输出

Sample1:

2

Sample2:

1

Sample3:

0

数组s[i]表示前i个数的前缀和，数组shu[i]表示前i个数的前缀和中值为s
*2/3的个数

#include<cstdio>
using namespace std;

long long s[500004],ans=0,shu[500004];

int main()
{
//freopen("a.txt","r",stdin);
//freopen("b1.txt","w",stdout);
int n;scanf("%d",&n);
long long a;
for(int i=1;i<=n;i++)
{
scanf("%I64d",&a);
s[i]=a+s[i-1];
}
if(n<3)
{
printf("%I64d",ans);
return 0;
}
if(s
%3!=0)
{
printf("%I64d",ans);
return 0;
}
long long h=s
/3;
for(int i=2;i<n;i++)
{
if(s[i]==h*2) shu[i]=shu[i-1]+1;
else shu[i]=shu[i-1];
}
for(int i=1;i<n-1;i++)
{
if(s[i]==h)
{
ans+=shu[n-1]-shu[i];
}
}
printf("%I64d",ans);
}