bzoj3771 Triple（生成函数+FFT+组合数学）

首先1个的直接统计
将所有的数搞成一个生成函数，做一遍卷积搞出来选2个的答案
但是2个的存在选了两个相同的，或者选了一个排列，直接除2即可
然后生成函数卷两次统计选3个的答案
这里需要容斥一下，(选3个的答案-强行选了2个一样的*3+强行选了3个一样的*2)/6才是不考虑顺序、选不重复的3个的答案
强行选了3个一样的直接枚举，强行选了2个一样的就将每一个数的两倍搞成生成函数再和1的卷一下求出
用FFT加速
摘自：portal

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <complex>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 40010
#define pi acos(-1)
typedef complex<double> E;
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,nn,L,R[N*12],val[N],ans[N*3],tmp[N*3];
E a[N*12],b[N*12];
inline void fft(E *a,int f){
    for(int i=0;i<n;++i) if(i>R[i]) swap(a[i],a[R[i]]);
    for(int i=1;i<n;i+=i){
        E wn(cos(pi/i),f*sin(pi/i));
        for(int j=0,p=i<<1;j<n;j+=p){
            E w(1,0);
            for(int k=0;k<i;++k,w*=wn){
                E x=a[j+k],y=a[j+k+i]*w;
                a[j+k]=x+y;a[j+k+i]=x-y;
            }
        }
    }if(f==-1) for(int i=0;i<n;++i) a[i]/=n;
}
int main(){
//  freopen("a.in","r",stdin);
    nn=n=read();for(int i=1;i<=n;++i) val[i]=read();
    //选1个
    for(int i=1;i<=n;++i) ans[val[i]]++;
    //选2个
    for(int i=1;i<=n;++i) a[val[i]]=1;
    m=val[n]*3;for(n=1;n<=m;n+=n) L++;
    for(int i=0;i<n;++i) R[i]=(R[i>>1]>>1)|(i&1)<<L-1;
    fft(a,1);for(int i=0;i<n;++i) b[i]=a[i]*a[i];fft(b,-1);
    for(int i=1;i<=val[nn]*2;++i) ans[i]+=(int)(b[i].real()+0.5)/2;//去重和相同
    //选3个
    for(int i=0;i<n;++i) b[i]=a[i]*a[i]*a[i];fft(b,-1);
    for(int i=1;i<=val[nn]*3;++i) tmp[i]+=(int)(b[i].real()+0.5);
    memset(b,0,sizeof(b));
    for(int i=1;i<=nn;++i) b[val[i]*2]=1;
    fft(b,1);for(int i=0;i<n;++i) b[i]*=a[i];fft(b,-1);
    for(int i=1;i<=val[nn]*3;++i) tmp[i]-=(int)(b[i].real()+0.5)*3;
    for(int i=1;i<=nn;++i) tmp[val[i]*3]+=2;
    for(int i=1;i<=val[nn]*3;++i) ans[i]+=tmp[i]/6;
    for(int i=1;i<=val[nn]*3;++i){
        if(!ans[i]) continue;
        printf("%d %d\n",i,ans[i]);
    }return 0;
}