codeforces 466C. Number of Ways 解题报告

题目链接：http://codeforces.com/problemset/problem/466/C

题目意思：给出一个 n 个数的序列你，问通过将序列分成三段，使得每段的和都相等的分法有多少种。

    这个是详细的题解：

    http://codeforces.com/blog/entry/13758

    

    代码也是按这个思路来做的，可怜我的做法改了 N 次总是得不到正确的结果～～泪

    

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;

typedef __int64 ll;

const int maxn = 5e5 + 5;
ll a[maxn], sum[maxn];
ll cnt[maxn];

int main()
{
    int n;
    while (scanf("%d", &n) != EOF)
    {
        memset(sum, 0, sizeof(sum));
        memset(cnt, 0, sizeof(cnt));

        for (int i = 1; i <= n; i++)
        {
            scanf("%I64d", &a[i]);
            sum[i] = sum[i-1] + a[i];
        }
        ll ans = 0;
        if (sum[n] % 3 == 0)   
        {
            ll avg = sum[n] / 3;

            for (int i = n-1; i; i--)
            {
                cnt[i] = cnt[i+1];    // 不断往前传递
                if (sum[i] == avg)
                    ans += cnt[i];
                if (sum[i] == avg * 2) // 可以分成三段的标志
                    cnt[i]++;
            }
        }
        printf("%I64d
", ans);
    }
    return 0;
}