标签：ref   time   include   tar   lan   swap   turn   scanf   方案   

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
*/

UVA12298 Super Poker II

标签：ref   time   include   tar   lan   swap   turn   scanf   方案   

原文地址：https://www.cnblogs.com/Troverld/p/14607976.html