XV.UVA12298 Super Poker II
我们设\(f_{i,j}\)表示遍历完前\(i\)种花色后，有多少种方案凑出和为\(j\)来。
再设\(g_{i,j}\)表示第\(i\)种花色是否存在点数为\(j\)的牌。
则有\(f_{i,j}=\sum\limits_{k=0}^jf_{i-1,k}\times g_{i,j-k}\)
明显是卷积形式，直接用FFT优化即可。
注意本题卡精度，需要用long double和long long。
代码：
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
const ld pi=acos(-1);
int n,a,b,rev[200100],lim,LG;
int arr[4][200100];
ll res[200100];
struct cp{
ld x,y;
cp(ld u=0,ld v=0){x=u,y=v;}
friend cp operator +(const cp &u,const cp &v){return cp(u.x+v.x,u.y+v.y);}
friend cp operator -(const cp &u,const cp &v){return cp(u.x-v.x,u.y-v.y);}
friend cp operator *(const cp &u,const cp &v){return cp(u.x*v.x-u.y*v.y,u.x*v.y+u.y*v.x);}
}A[200100],B[200100];
void FFT(cp *a,int tp){
for(int i=0;i<lim;i++)if(i<rev[i])swap(a[i],a[rev[i]]);
for(int md=1;md<lim;md<<=1){
cp rt=cp(cos(pi/md),tp*sin(pi/md));
for(int stp=md<<1,pos=0;pos<lim;pos+=stp){
cp w=cp(1,0);
for(int i=0;i<md;i++,w=w*rt){
cp x=a[pos+i],y=w*a[pos+md+i];
a[pos+i]=x+y;
a[pos+md+i]=x-y;
}
}
}
if(tp==-1)for(int i=0;i<lim;i++)a[i].x/=lim;
}
void mul(ll *a,int *b){
for(int i=0;i<lim;i++)A[i]=B[i]=cp(0,0);
for(int i=0;i<(lim>>1);i++)A[i]=cp(a[i],0),B[i]=cp(b[i],0);
FFT(A,1),FFT(B,1);
for(int i=0;i<lim;i++)A[i]=A[i]*B[i];
FFT(A,-1);
for(int i=0;i<lim;i++)a[i]=(ll)(A[i].x+0.5);
}
bool comp(int x){
for(int i=2;i*i<=x;i++)if(!(x%i))return true;
return false;
}
int main(){
scanf("%d%d%d",&a,&b,&n);
while(a||b||n){
lim=1,LG=0,b++;
while(lim<=(b<<1))lim<<=1,LG++;
for(int i=0;i<lim;i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(LG-1));
for(int i=0;i<lim;i++)for(int j=0;j<4;j++)arr[j][i]=comp(i);
for(int i=0;i<lim;i++)res[i]=0;
res[0]=1;
for(int i=1,x,y;i<=n;i++){
char c;
scanf("%d%c",&x,&c);
if(c=='S')y=0;
if(c=='H')y=1;
if(c=='C')y=2;
if(c=='D')y=3;
arr[y][x]=0;
}
// for(int i=0;i<4;i++){for(int j=0;j<lim;j++)printf("%d ",arr[i][j]);puts("");}
for(int i=0;i<4;i++)mul(res,arr[i]);
for(int i=a;i<b;i++)printf("%lld\n",res[i]);
putchar('\n');
scanf("%d%d%d",&a,&b,&n);
}
return 0;
}
/*
12 30 10
14S 15H 12C 6S 4D 6H 15C 9D 22C 18S
0 0 0
*/