Description
You've got array a[1], a[2], ..., a[n], consisting of n integers. Count the number of ways to split all the elements of the array into three contiguous parts so that the sum of elements in each part is the same.
More formally, you need to find the number of such pairs of indices i, j(2 ≤ i ≤ j ≤ n - 1), that
.
Input
The first line contains integer n(1 ≤ n ≤ 5·105), showing how many numbers are in the array. The second line contains n integersa[1], a[2], ..., a[n](|a[i]| ≤ 109) — the elements of array a.
Output
Print a single integer — the number of ways to split the array into three parts with the same sum.
Sample Input
5 1 2 3 0 3
2
4 0 1 -1 0
1
2 4 1
0
小结:
....题目看清楚才发现这是一道彻彻底底的水题,原来以为要用DP做的...
以下是AC代码:
#include <iostream>
#include <cstdio>
typedef long long ll;
using namespace std;
const int maxn = 500005;
ll sum[maxn];
int main()
{
int n;
scanf("%d", &n);
sum[0] = 0;
for (int i = 0; i < n; i++)
{
int x;
scanf("%d", &x);
sum[i+1] = sum[i] + x;
}
ll ans = 0;
if (sum[n] % 3 == 0)
{
ll one = sum[n] / 3, two = sum[n] / 3 * 2;
ll cnt = 0;
for (int i = 1; i < n; i++)
{
if (sum[i] == two)
ans += cnt;
if (sum[i] == one)
cnt++;
}
}
cout << ans << endl;
return 0;
}
本文介绍了一个算法,用于计算数组中三个连续部分的元素和相等的情况数量。通过输入数组长度和数组元素,输出满足条件的组合数量。

被折叠的 条评论
为什么被折叠?



