BZOJ 3992: [SDOI2015]序列统计 [快速数论变换 生成函数 离散对数]

#include <iostream>
#include <cstdio>
#include <string>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int N=1e5+5;
inline int read(){
    char c=getchar();int x=0,f=1;
    while(c<‘0‘||c>‘9‘){if(c==‘-‘)f=-1;c=getchar();}
    while(c>=‘0‘&&c<=‘9‘){x=x*10+c-‘0‘;c=getchar();}
    return x*f;
}
int n,m,x,S;
ll A[N],ans[N];
 
ll P=1004535809,MOD=P;
ll Pow(ll a,ll b,ll MOD){
    ll ans=1;
    for(;b;b>>=1,a=a*a%MOD)
        if(b&1) ans=ans*a%MOD;
    return ans;
}
ll PrimitiveRoot(ll p){
    if(p==2) return 1;
    for(ll g=2;g<p;g++){
        bool flag=1;ll m=sqrt(p);
        for(ll i=2;i<=m;i++) if((p-1)%i==0)
            if(Pow(g,(p-1)/i,p)==1) {flag=0;break;}
        if(flag) return g;
    }
    return 0;
}
int ind[N];
void iniInd(){
    int g=PrimitiveRoot(m),a=1;
    for(int i=0;i<m-1;i++,a=a*g%m) ind[a]=i;
}
 
struct NumberTheoreticTransform{
    int n,rev[N];
    ll g;
    void ini(int m){
        n=1;
        while(n<m) n<<=1;
         
        int k=0;
        while((1<<k)<n) k++;
        for(int i=0;i<n;i++){
            int t=0;
            for(int j=0;j<k;j++) if(i&(1<<j)) t|=(1<<(k-j-1));
            rev[i]=t;
        }
 
        g=3;
    }
    void DFT(ll *a,int flag){
        for(int i=0;i<n;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
        for(int l=2;l<=n;l<<=1){
            int m=l>>1;
            ll wn=Pow(g,flag==1?(P-1)/l:P-1-(P-1)/l,P);
            for(ll *p=a;p!=a+n;p+=l){
                ll w=1;
                for(int k=0;k<m;k++){
                    ll t=w*p[k+m]%P;
                    p[k+m]=(p[k]-t+P)%P;
                    p[k]=(p[k]+t)%P;
                    w=w*wn%P;
                }
            }
        }
        if(flag==-1){
            ll inv=Pow(n,P-2,P);;
            for(int i=0;i<n;i++) a[i]=a[i]*inv%P;
        }
    }
    void SQR(ll *A){
        DFT(A,1);
        for(int i=0;i<n;i++) A[i]=A[i]*A[i]%MOD;
        DFT(A,-1);
        for(int i=0;i<=m-2;i++) 
            A[i]=(A[i]+A[i+m-1])%MOD,A[i+m-1]=0;
    }
    ll C[N];
    void MUL(ll *A,ll *B){
        for(int i=0;i<n;i++) C[i]=B[i];
        DFT(A,1);DFT(C,1);
        for(int i=0;i<n;i++) A[i]=A[i]*C[i]%MOD;
        DFT(A,-1);
        for(int i=0;i<=m-2;i++) 
            A[i]=(A[i]+A[i+m-1])%MOD,A[i+m-1]=0;
    }
    void PowPoly(ll *A,int b,ll *ans){
        ans[0]=1;
        for(;b;b>>=1,SQR(A))
            if(b&1) MUL(ans,A);
    }
}fft;
     
int main(){
    freopen("in","r",stdin);
    n=read();m=read();x=read();S=read();
    fft.ini(m+m);
    iniInd();
    for(int i=1;i<=S;i++){
        int x=read();
        if(x) A[ind[x]]=1;
    }
 
    fft.PowPoly(A,n,ans);
    printf("%lld",ans[ind[x]]);
}