题意
给定n个数,求从其中选1个、2个、3个能得到的权值和的情况及方案数
生成函数裸题
刚接触这玩意并不是很能理解,也不是很懂怎么表达……
考虑
f(i)
表示权值和为i的方案数,那么有
f(i)=∑f(j)∗f(i−j)
,为卷积形式,
那么令母函数
g(x)=∑aixi
的第m项系数为权值和为m的方案数,就可以用FFT优化多项式乘法来解决。
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <complex>
#define N 400010
using namespace std;
typedef complex<double> E;
typedef long long ll;
int n,x,L,m;
E A[N],B[N],C[N],Ans[N];
int rev[N];
inline void reaD(int &x){
char Ch=getchar();x=0;
for(;Ch>'9'||Ch<'0';Ch=getchar());
for(;Ch>='0'&&Ch<='9';x=x*10+Ch-'0',Ch=getchar());
}
inline void FFT(E *a,int w){
for(int i=0;i<n;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
for(int i=1;i<n;i<<=1){
E wn(cos(M_PI/i),w*sin(M_PI/i));
for(int j=0;j<n;j+=(i<<1)){
E w(1,0);
for(int k=0;k<i;k++,w*=wn){
E x=a[j+k],y=w*a[j+k+i];
a[j+k]=x+y; a[j+k+i]=x-y;
}
}
}
if(w==-1) for(int i=0;i<n;i++) a[i]/=n;
}
int main(){
reaD(n);
for(int i=1;i<=n;i++){
reaD(x);
A[x]+=1; B[2*x]+=1; C[3*x]+=1;
m=max(m,3*x);
}
m<<=1; m++;
for(n=1,L=0;n<m;n<<=1)L++;
for(int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(L-1));
FFT(A,1); FFT(B,1); FFT(C,1);
E six=6,three=3,two=2;
for(int i=0;i<n;i++)
Ans[i]=(A[i]*A[i]*A[i]-three*A[i]*B[i]+two*C[i])/six+(A[i]*A[i]-B[i])/two+A[i];
FFT(Ans,-1);
for(int i=1;i<m;i++)
if((int)(Ans[i].real()+0.5)){
printf("%d %d\n",i,(int)(Ans[i].real()+0.5));
}
return 0;
}