1. 程式人生 > >12298-Super Poker II (FFT)

12298-Super Poker II (FFT)

大致題意:有一副撲克,有無數張牌。對於每個正合數p,恰有4張牌,黑桃p, 紅桃p, 方塊p。 現在告訴你,有些牌已經丟失了。 給定兩個數a,b 求區間[a,b]中的每個整數x, 從4種花色中各選一張牌,問有多少種組合可以使得點數之和等於x

題解:對於每種花色的牌,可以將其看做一個多項式,對於缺失的牌令其對應係數為0即可,然後就是四次多項式乘法了,很明顯一定要用FFT優化的,但是這題太卡精度了,全部改成long double才能過。還是待趕緊學會NTT。。。

#include<math.h>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
const long double PI = acos(-1.0);
#define ll long long 
#define maxn 50100
ll a[maxn]={1,1},b[maxn],cnt;
void init()
{
	b[0]=b[1]=1;
	for(ll i=2;i<maxn-10;i++)
	{
		if(a[i])
			continue;
		b[i]=1;
		for(ll j=i;j<maxn-10;j+=i)
			a[j]=1;
	}
	//for(int i=1;i<=10;i++)
		//if(!b[i])
			//printf("%d\n",i);
}
struct complex 
{
	long double r,i;
	complex(long double _r=0.0,long double _i=0.0)
	{
		r=_r;i=_i;
	}
	complex operator +(const complex &b)
	{
		return complex(r+b.r,i+b.i);
	}
	complex operator -(const complex &b)
	{
		return complex(r-b.r,i-b.i);
	}
	complex operator *(const complex &b)
    {
        return complex(r*b.r-i*b.i,r*b.i+i*b.r);
    }
};
/*
 * 進行FFT和IFFT前的反轉變換。
 * 位置i和 (i二進位制反轉後位置)互換
 * len必須取2的冪
 */
complex s[1<<19],c[1<<19],d[1<<19],h[1<<19];
int A,B,C;
//雷德演算法--倒位序  
void change(complex y[],int len)
{
	int i,j,k;
	for(i=1,j=len/2;i<len-1;i++)
	{
		if(i<j)            
			swap(y[i],y[j]);
		//交換互為小標的元素,i<j保證只交換一次
		//i做正常的+1,j左反轉型別的+1,始終保持i和j是反轉的
		k=len/2;
		while(j>=k)
			j-=k,k/=2;
		if(j<k)
			j+=k;  
	}
}
/*
 * 做FFT
 * len必須為2^k形式,
 * on==1時是DFT,on==-1時是IDFT
 */
void fft(complex y[],int len,int on)
{
	change(y,len);
	for(int h=2;h<=len;h<<=1)//分治後計算長度為h的DFT
	{
		complex wn(cos(-on*2*PI/h),sin(-on*2*PI/h));//單位復根e^(2*PI/m)用尤拉公式展開 
		for(int j=0;j<len;j+=h)
		{
			complex w(1,0);//旋轉因子  
			for(int k=j;k<j+h/2;k++)
			{
				complex u=y[k];
				complex t=w*y[k+h/2];
				y[k]=u+t;//蝴蝶合併操作 
				y[k+h/2]=u-t;
				w=w*wn;//更新旋轉因子 
			}
		}
	}
	if(on == -1)
        for(int i = 0;i < len;i++)
            y[i].r /= len;
}
int main(void)
{
	init();
	
    while(scanf("%d %d %d", &A, &B, &C), A || B || C)
	{
		int len=1;
		while(len<=B) 
			len<<=1;len<<=3;
		for(int i=0;i<=len;i++)
			s[i]=h[i]=c[i]=d[i]=complex(0,0);
		for(int i=0;i<=B;i++)
			if(!b[i])
				s[i]=h[i]=c[i]=d[i]=complex(1,0);
		for(int i=1;i<=C;i++)
		{
			char ss;int t;
			scanf("%d%c",&t,&ss);
			if(ss=='S') s[t].r=0;
			if(ss=='H') h[t].r=0;
			if(ss=='C') c[t].r=0;
			if(ss=='D') d[t].r=0;
		}
		fft(s,len,1);fft(h,len,1);
		fft(c,len,1);fft(d,len,1);
		for(int i=0;i<len;i++)
			h[i]=h[i]*s[i]*c[i]*d[i];
		fft(h,len,-1);
		for(int i=A;i<=B;i++)
            printf("%lld\n", (ll)(h[i].r+0.5));
		puts("");
	}
	return 0;
}