NKOJ 3049 三分数组
时间限制 : 10000 MS 空间限制 : 65536 KB
问题描述
给出一个有n 个整数的数组a[1],a[2],…,a[n], 有多少种方法把数组分成3 个连续的子序列,使得各子序列的元素之和相等。也就是说,有多少个下标对i,j (2≤i≤j≤n-1), 满足:sum(a[1]..a[i-1]) = sum(a[i]..a[j]) = sum(a[j+1]..a[n])
输入格式
第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[n]*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[n]%3!=0)
{
printf("%I64d",ans);
return 0;
}
long long h=s[n]/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);
}