[bzoj4927][SDOI省队集训2017]diyiti

题目大意

在一个长度为n的数列中找出六个数，使得可以把它们分成四组，每组的和均相等。求方案数。

XJB做

只有1+1+1+3或1+1+2+2两种拆分。
先预处理two[x]表示找出两个数和为x的方案数，以及one[x]表示有多少数是x。
对于1+1+1+3，枚举那个1是x，首先有一个$C_{one[x]}^3$。
对于那个3，我们先枚举一个ai，然后对应方案数是two[x-a[i]]。
注意当a[i]+a[i]+a[j]=x时会记重，要减去。
然后还会出现很尴尬得a[i]+a[i]+a[i]=x的情况。
大概讨论下。
对于1+1+2+2也是讨论，突然不想写了请看代码。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=5000+10,maxd=10000000+10;
int a[maxn],one[maxd],two[maxd];
bool bz[maxd];
int i,j,k,l,t,n,m,x,z;
ll w,y,ans;
ll C(int n,int m){
    if (m==2) return (ll)n*(n-1)/2;
    else if (m==3) return (ll)n*(n-1)*(n-2)/6;
}
int main(){
    freopen("yist.in","r",stdin);freopen("yist.out","w",stdout);
    scanf("%d",&n);
    fo(i,1,n){
        scanf("%d",&a[i]);
        one[a[i]]++;
    }
    fo(i,1,n-1)
        fo(j,i+1,n)
            if (a[i]+a[j]<=maxd-10) two[a[i]+a[j]]++;
    fo(i,1,n){
        if (bz[a[i]]||one[a[i]]<3) continue;
        bz[a[i]]=1;
        x=a[i];
        y=0;
        fo(j,1,n)
            if (x>=a[j]){
                z=0;
                if (2*a[j]<=x){
                    z=one[x-2*a[j]];
                    if (3*a[j]==x) z--;
                }
                y+=(ll)(two[x-a[j]]-z);
            }
        y/=3;
        ans+=(ll)C(one[x],3)*y;
    }
    fo(i,1,n) bz[a[i]]=0;
    fo(i,1,n){
        if (bz[a[i]]||one[a[i]]<2) continue;
        bz[a[i]]=1;
        x=a[i];
        y=C(two[x],2);
        fo(j,1,n)
            if (x>=a[j]){
                z=one[x-a[j]];
                if (2*a[j]==x) z--;
                y-=(ll)C(z,2);
            }
        w=0;
        fo(j,1,n)
            if (a[j]*2<x) w+=(ll)(one[a[j]]-1)*C(one[x-a[j]],2);
        w/=2;
        y-=w;
        w=0;
        fo(j,1,n)
            if (a[j]*2==x) w+=(ll)C(one[a[j]]-1,3)*2;
        w/=4;
        y-=w;
        ans+=(ll)C(one[x],2)*y;
    }
    printf("%lld\n",ans);
}