BZOJ3771：Triple（生成函数+FFT+容斥）

题面
题意：一些数，对于每个x，选3个以下的数，不计顺序，和为x的方案数。

顺序就是阶乘，先不管，最后乘上就行了

设生成函数为A
考虑选两个数，直接卷积后会算重复的，减去两个相同的方案数
三个的话，会算重选了两个一样的
减去后还要加上三个一样的
可以强行对于每种都选了两个一样的，再做一个生成函数B
选三个一样的，生成函数C

容斥一下发现是大概这样的
$A^3-3A*B+2*C$
FFT加速就行了

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

const double pi=acos(-1);
const int N=144444;

int n,rev[N];

struct yy
{
    double x,y;
    yy(double a=0,double b=0):x(a),y(b){}
};

yy operator +(yy a,yy b){return yy(a.x+b.x,a.y+b.y);}
yy operator -(yy a,yy b){return yy(a.x-b.x,a.y-b.y);}
yy operator *(yy a,yy b){return yy(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}

void init(int lim)
{
    int k=-1;
    n=1;
    while(n<=lim)
    n<<=1,k++;
    for(int i=0;i<n;i++)
    rev[i]=(rev[i>>1]>>1) | ((i&1)<<k);
}

void fft(yy *a,int ops)
{
    for(int i=0;i<n;i++)
    if(i<rev[i])
    swap(a[i],a[rev[i]]);

    for(int m=1,l=2;l<=n;m<<=1,l<<=1)
    {
        yy wn=yy(cos(pi/m),ops*sin(pi/m));
        for(int i=0;i<n;i+=l)
        {
            yy w=yy(1,0);
            for(int k=0;k<m;k++,w=w*wn)
            {
                yy t=a[i+k+m]*w;
                a[i+k+m]=a[i+k]-t;
                a[i+k]=a[i+k]+t;
            }
        }
    }
    if(ops==-1)
    for(int i=0;i<n;i++)
    a[i].x/=n;
}

yy a[N],b[N],c[N];
yy aa[N];
LL ans[N];

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        int hy;
        scanf("%d",&hy);
        ans[hy]++;
        a[hy].x+=1.0;
        b[hy+hy].x+=1.0;
        c[hy+hy+hy].x+=1.0;
    }
    init(120010);
    fft(a,1);
    for(int i=0;i<n;i++)
    aa[i]=a[i]*a[i];
    fft(aa,-1);

    for(int i=0;i<n;i++)
    ans[i]+=(LL)(aa[i].x-b[i].x+0.4)/2;

    for(int i=0;i<n;i++)
    aa[i]=a[i]*a[i]*a[i];
    fft(aa,-1);

    fft(b,1);
    for(int i=0;i<n;i++)
    b[i]=b[i]*a[i];
    fft(b,-1);

    for(int i=0;i<n;i++)
    ans[i]+=(LL)(aa[i].x-3*b[i].x+2*c[i].x+0.4)/6;

    for(int i=0;i<n;i++)
    if(ans[i]!=0)
    printf("%d %lld\n",i,ans[i]);

    return 0;
}