[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，首先有一个C3one[x]。
对于那个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);}