1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#include <cstdio>
#include <map>
#define ll long long
using namespace std;
ll a[100005], sum[100005];
int main() {
    int n;
    scanf("%d", &n);
    map<ll, ll> sum_cnt;
    for (int i = 1; i <= n; ++i)
        scanf("%lld", &a[i]), sum[i] = sum[i - 1] + a[i];
    if (sum[n] < 0)
        for (int i = 1; i <= n; ++i)
            a[i] = -a[i], sum[i] = -sum[i], sum_cnt[sum[i]]++;
    else
        for (int i = 1; i <= n; ++i) sum_cnt[sum[i]]++;
    ll temp = sum[n] / 3;
    if (sum[n] % 3 == 0) {
        ll sum1 = 0, ans = 0;
        sum_cnt[sum[n]]--;
        for (int i = n; i > 2; --i) {
            sum1 += a[i];
            sum_cnt[sum[i - 1]]--;
            if (sum1 == temp) {
                ans += sum_cnt[temp];
            }
        }
        printf("%lld\n", ans);
    } else if (sum[n] % 3 == 1) {
        ll sum1 = 0, ans = 0;
        sum_cnt[sum[n]]--;
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX