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 + ;
 ll a[maxn], sum[maxn];
 ll cnt[maxn];

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

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

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