FFT－Super Poker II－UVA - 12298

FFT－Super Poker II－UVA - 12298

题意：

$$\begin{gathered} 有一副扑克，包含无数张牌。对于每个正合数p，恰好又4张牌：黑桃p，红桃p， \\ 梅花p和方块p(分别用pS、pH、pC和pD表示)。没有其他类型的牌。 \end{gathered}$$

$$\begin{gathered} 给定一个正整数n，从4中花色中各选一张牌，问有多少种组合数可以使得点数之和等于n。 \\ 例如，n=24时，有一种组合方法是4S+6H+4C+10D。如下图所示： \end{gathered}$$

$但是有c张牌丢失了。$

$给定两个正整数a和b，要求n=a,n=a+1,...,n=b时的答案。$

输入格式

$输入包含不超过25组数据。$

$每组第一行包含3三个正整数a,b,c$

$第二行包含c个不同的字符串，即已丢失的牌。$

$这些牌型如pS、pH、pC或者pD，其中p是一个正合数$

$输入结束标志为a=b=c=0$

输出格式

$对每组数据，输出b-a+1行，每行一个整数。每组数据后输出一个空行。$

数据范围：

$最多一组数据满足a=1,b=50000且c\le 10000，其他数据1\le a\le b\le 100，0\le c\le 10$

---

分析：

$用多项式解决此类组合问题。$

$用四个多项式表示4种牌的选法集合。$

$每个多项式对应为：1+a_1\cdot x^4+a_2\cdot x^6+...+a_k\cdot x^p，$

$当第i个合数p_i可选择时，a_i=1，否则a_i=0，$

$我们将这4个多项式相乘，最终的结果就是指数为p的多项式的系数。$

$具体参见刘汝佳白书P430。$

注意：

$①、最后输出时，要先将结果强转为longlong，再输出，否则精度会出现问题。$

$②、多项式的长度根据多项式最终乘积的最高次数来确定。$

$③、a、b、c均为0，在a,b,c\ge 0的情况下，判断条件是a+b+c=0。$

$④、PI要开longdouble，精度卡的太严格了！$

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<iostream>
#include<cmath>

#define ll long long

using namespace std;

const long double PI = acos(-1.0);

struct Complex
{
    long double x, y;
    Complex(long double _x = 0.0, long double _y = 0.0)
    {
        x = _x;
        y = _y;
    }
    Complex operator - (const Complex &b) const
    {
        return Complex(x - b.x, y - b.y);
    }
    Complex operator + (const Complex &b) const
    {
        return Complex(x + b.x, y + b.y);
    }
    Complex operator * (const Complex &b) const
    {
        return Complex(x*b.x - y*b.y, x*b.y + y*b.x);
    }
};

void change(Complex y[], int len)
{
    int i, j, k;
    for(int i=1, j=len/2; i<len-1; i++)
    {
        if(i<j) swap(y[i], y[j]);
        k = len/2;
        while(j>=k)
        {
            j -= k;
            k /= 2;
        }
        if(j<k) j += k;
    }
}

void fft(Complex y[], int len, int on)
{
    change(y, len);
    for(int h = 2; h<=len; h<<=1)
    {
        Complex wn(cos(-on*2*PI/h), sin(-on*2*PI/h));
        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].x /= len;
}

const int N = 5e4+10;

Complex x[4][N*16];		//最多四个多项式相乘
int a,b,c;
char str[10];
int primes[N], cnt;
bool st[N];

void get_prime(int n)
{
    for(int i=2;i<=n;i++)
    {
        if(!st[i]) primes[cnt++]=i;
        for(int j=0;primes[j]*i<=n;j++)
        {
            st[i*primes[j]]=true;
            if(i%primes[j]==0) break;
        }
    }
}

int main()
{
    get_prime(N-1);
    while(~scanf("%d%d%d",&a,&b,&c), a + b + c)
    {
        int len = 1;
        while(len <= b) len <<= 1;         //有4个多项式乘积，次数最多为4b，
        len <<= 3;							//即n=4b，则2n=8b
        
        for(int j=0;j<4;j++)
            for(int i=0;i<=b;i++) 
                if(st[i]) x[j][i] = Complex(1,0);
                else x[j][i] = Complex(0,0);    //多测 要清空
        for(int j=0;j<4;j++)
            for(int i=b+1;i<len;i++)
                x[j][i] = Complex(0,0);     //虽然没有必要，但是要注意步骤
        
        for(int i=0;i<c;i++)
        {
            scanf("%s",str);
            int k = strlen(str), id;
            if(str[k-1]=='S') id = 0;
            else if(str[k-1]=='H') id = 1;
            else if(str[k-1]=='C') id = 2;
            else id = 3;
            str[k-1]='\0';
            int m = atoi(str);
            x[id][m] = Complex(0,0);
        }

        for(int i=0;i<4;i++) fft(x[i], len, 1);
        
        for(int i=1;i<4;i++) 
            for(int j=0;j<len;j++)
                x[0][j] = x[0][j]*x[i][j];
            
        fft(x[0], len, -1);

        for(int i=a;i<=b;i++) printf("%lld\n",(ll)(x[0][i].x+0.5));
        puts("");
    }
    return 0;
}