【普通型母函数+容斥+FFT】BZOJ3771[Triple]题解

题目概述

ZigZagK有 $n$ 把价值不一样的斧子，CHNJZ偷走了 $1$ 把或 $2$ 把或 $3$ 把，对于每个可能的总损失，计算有几种可能的方案。

解题报告

emm……显然是母函数啊？但是有数量限制。由于最多偷走三把，所以我们可以直接三种情况都讨论过去。

$1$ 把：母函数：$A(x)=x^{a_1}+x^{a_2}+x^{a_3}+\cdots+x^{a_n}$ 。

$2$ 把：母函数：$A^2(x)$ ，但是这样显然算进了同把斧子用了两次的情况，所以要去掉 $B(x)=x^{2a_1}+x^{2a_2}+x^{2a_3}+\cdots+x^{2a_n}$ 。又因为这样算进了排列，所以最后要除以 $2$ 。

$3$ 把：母函数：$A^3(x)$ ，去掉 $3A(x)B(x)$ ，补回 $2[C(x)=x^{3a_1}+x^{3a_2}+x^{3a_3}+\cdots+x^{3a_n}]$ ，除以 6 <script type="math/tex" id="MathJax-Element-15">6</script> 。

因为项数比较多，所以用FFT进行多项式乘法。

解题报告

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fr first
#define sc second
#define mp make_pair
using namespace std;
typedef long double DB;typedef pair<DB,DB> CN;
const int maxn=262144;const DB pi=acos(-1);

int n,a[maxn+5],MAX,R[maxn+5],ans[maxn+5];
CN A[maxn+5],B[maxn+5],C[maxn+5],now[maxn+5];

inline void Rev(int n){
    int len=-1;for (int x=1;x<n;x<<=1) len++;R[0]=0;
    for (int i=1;i<n;i++) R[i]=(R[i>>1]>>1)|((i&1)<<len);
}
inline CN operator + (const CN &a,const CN &b) {return mp(a.fr+b.fr,a.sc+b.sc);}
inline CN operator - (const CN &a,const CN &b) {return mp(a.fr-b.fr,a.sc-b.sc);}
inline CN operator * (const CN &a,const CN &b) {return mp(a.fr*b.fr-a.sc*b.sc,a.fr*b.sc+a.sc*b.fr);}
inline void FFT(CN *a,int n,int f){
    for (int i=0;i<n;i++) if (i<R[i]) swap(a[i],a[R[i]]);
    for (int k=1;k<n;k<<=1){
        CN w=mp(1,0),wn=mp(cos(pi/k),sin(f*pi/k)),x,y;
        for (int i=0;i<n;i+=k<<1,w=mp(1,0))
            for (int j=0;j<k;j++,w=w*wn)
                x=a[i+j],y=w*a[i+j+k],a[i+j]=x+y,a[i+j+k]=x-y;
    }
    if (f<0) for (int i=0;i<n;i++) a[i].fr/=n;
}
int main(){
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    scanf("%d",&n);for (int i=1;i<=n;i++) scanf("%d",&a[i]),MAX=max(MAX,a[i]);
    for (int i=1;i<=n;i++) A[a[i]]=B[a[i]*2]=C[a[i]*3]=mp(1,0);MAX*=3;
    for (n=1;n<=(MAX<<1);n<<=1);Rev(n);for (int i=0;i<n;i++) ans[i]+=A[i].fr+0.1;
    FFT(A,n,1);for (int i=0;i<n;i++) now[i]=A[i]*A[i];FFT(now,n,-1);
    for (int i=0;i<n;i++) ans[i]+=(now[i].fr-B[i].fr)/2+0.1;
    FFT(B,n,1);for (int i=0;i<n;i++) B[i]=A[i]*B[i];FFT(B,n,-1);
    for (int i=0;i<n;i++) now[i]=A[i]*A[i]*A[i];FFT(now,n,-1);
    for (int i=0;i<n;i++) ans[i]+=(now[i].fr-B[i].fr*3+C[i].fr*2)/6+0.1;
    for (int i=0;i<=MAX;i++) if (ans[i]) printf("%d %d\n",i,ans[i]);
    return 0;
}