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

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);
}