Codeforces Round #266 (Div. 2) C

来源：互联网发布：第四类接触知乎编辑：程序博客网时间：2024/05/21 08:58

C. Number of Ways

题意：一个数列a1~an，需要把它分割成连续的三段，且每段的和相等。求有多少种分法。

思路：开一个数组存前i项和，如果数列的和能被3整除的话，算出总和sum(n)的1/3和2/3。扫一遍前i项和，如果发现sum(i)值为总和的1/3，则计数；如果发现sum(i)的值为总和的2/3，那么说明以i结尾作第二段分割的分法等于1~i中sum(x)==1/3*sum(n)的数量。

#include <iostream>             #include <stdio.h>             #include <cmath>             #include <algorithm>             #include <iomanip>             #include <cstdlib>             #include <string>             #include <memory.h>             #include <vector>             #include <queue>             #include <stack>             #include <map>           #include <set>           #include <ctype.h>           #define INF 10000000000LL        #define ll long long         #define max3(a,b,c) max(a,max(b,c))         using namespace std;ll num[500010];ll sum[500010];int main(){int n;while(cin>>n){ll cnt=0;ll ans=0;sum[0]=0;for(int i=1;i<=n;i++){scanf("%I64d",&num[i]);sum[i]=sum[i-1]+num[i];}if((sum[n]%3)==0){ll tri=sum[n]/3;for(int i=1;i<n;i++){if(i!=1&&sum[i]==2*tri){ans+=cnt;}if(sum[i]==tri)cnt++;}cout<<ans<<endl;}else{cout<<0<<endl;}}return 0;}

0 0