Codeforces - 466C - Number of Ways - 组合数学

https://codeforces.com/problemset/problem/466/C

要把数据分为均等的非空的三组，那么每次确定第二个分割点的时候把（除此之外的）第一个分割点的数目加上就可以了。记得最后给第三组留至少一个。

#include<bits/stdc++.h>
using namespace std;
#define ll long long

int n;
int a[];

int main(){
    scanf("%d",&n);
    for(int i=;i<n;i++)
        scanf("%d",&a[i]);

    ll sum=;
    for(int i=;i<n;i++){
        sum+=a[i];
    }

    if(sum%!=){
        puts("");
        return ;
    }

    ll d=sum/;
    int z1=;

    ll cnt=;

    ll cur=;
    int inc=;

    int i=;
    while(i<n){
        cur+=a[i];
        if(cur==d){
            z1++;
            inc=;
        }
        else{
            inc=;
        }

        if(z1>=){
            if(cur==2ll*d){
                if(i<=n-){
                    //至少把最后一个n-1留给第三组
                    if(inc)
                        cnt+=z1-;
                    else
                        cnt+=z1;
                    //printf("cnt=%lld i=%d\n",cnt,i);
                }
            }
        }
        i++;
    }

    printf("%lld\n",cnt);
}