【多项式快速幂】bzoj3992: [SDOI2015]序列统计

最新推荐文章于 2023-05-26 20:43:09 发布

RZXZ

最新推荐文章于 2023-05-26 20:43:09 发布

阅读量613

点赞数

分类专栏：多项式文章标签： bzoj3992

本文链接：https://blog.csdn.net/acmer_cheer/article/details/51258058

版权

多项式专栏收录该内容

1 篇文章 0 订阅

订阅专栏

题目链接

如果对生成函数稍有了解的，应该可以一眼看出这题要用生成函数吧…
附上蒟蒻对生成函数浅层次的理解：

f = a 0 g 0 + a 1 g 1 + a 2 g 2 + . . .

$f=a_0g^0+a_1g^1+a_2g^2+...$
其中

ai $a_i$ 取决于达到

i <script type="math/tex" id="MathJax-Element-3">i</script>的状态的个数。
转移的时候就变成了两个多项式相乘，就可以用FFT，NTT之类的东西辣。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#define LL long long int
#define MAXN 17005
using namespace std;

const int mod=(479<<21)+1, g=3, T=16384;
int n, m, x, t, h[MAXN], r[MAXN], len, pos[MAXN];
LL wn[MAXN], invwn[MAXN], invn;

LL power(LL a,LL pos)
{
    LL ans=1;
    for(;pos;pos>>=1, a=a*a%mod)
        if(pos&1)ans=ans*a%mod;
    return ans;
}

void init()
{
    for(int i=0;i<=(1<<len);++i)r[i]=(r[i>>1]>>1)|((i&1)<<(len-1));
    len=1<<len;
    LL r=power(g,(mod-1)/T), invr=power(r,mod-2);
    wn[0]=invwn[0]=1;
    for(int i=1;i<T;++i)
        wn[i]=wn[i-1]*r%mod, invwn[i]=invwn[i-1]*invr%mod;
    invn=power(len,mod-2);
}

int vis[MAXN], tim;
bool check(int x)
{
    ++tim;
    for(int i=1, now=1;i<m;++i, now=now*x%m)
    {
        if(vis[now]==tim)return 0;
        vis[now]=tim;
    }
    return 1;
}
int findroot(){for(int i=2;i<=m;++i)if(check(i))return i;return 0;}

struct Vector
{
    LL num[MAXN];
    Vector(){memset(num,0,sizeof num);}
    void NTT(int f)
    {
        for(int i=0;i<len;++i)if(i<r[i])swap(num[i],num[r[i]]);
        LL *a=f==1?wn:invwn, x, y;
        for(int i=1;i<len;i<<=1)
            for(int j=0;j<len;j+=i<<1)
                for(int k=0;k<i;++k)
                {
                    x=num[j+k], y=a[T/(i<<1)*k]*num[i+j+k]%mod;
                    num[j+k]=(x+y)%mod, num[i+j+k]=(x-y+mod)%mod;
                }
        if(f==-1)for(int i=0;i<len;++i)num[i]=num[i]*invn%mod;
    }
}a;

Vector qpow(Vector &a,LL pos)
{
    Vector ans;
    ans.num[0]=1;
    for(;pos;pos>>=1)
    {
        a.NTT(1);
        if(pos&1)
        {
            ans.NTT(1);
            for(int i=0;i<len;++i)ans.num[i]=ans.num[i]*a.num[i]%mod;
            ans.NTT(-1);
            for(int i=len-1;i>m;--i)ans.num[i-m+1]=(ans.num[i-m+1]+ans.num[i])%mod, ans.num[i]=0;
        }
        for(int i=0;i<len;++i)a.num[i]=a.num[i]*a.num[i]%mod;
        a.NTT(-1);
        for(int i=len-1;i>m;--i)a.num[i-m+1]=(a.num[i-m+1]+a.num[i])%mod, a.num[i]=0;
    }
    return ans;
}

int main()
{
    scanf("%d%d%d%d",&n,&m,&x,&t);
    for(int i=1;i<=t;++i)scanf("%d",&h[i]);

    len=(int)ceil(log2(m))+1;
    init();
    for(int i=0, root=findroot(), res=1;i<m-1;++i, res=res*root%m)
        pos[res]=i;
    for(int i=1;i<=t;++i)if(h[i])++a.num[pos[h[i]]];

    cout<<qpow(a,n).num[pos[x]]<<endl;
    return 0;
}