【BZOJ3992】【SDOI2015】序列统计

数论劲啊

原题：

小C有一个集合S，里面的元素都是小于M的非负整数。他用程序编写了一个数列生成器，可以生成一个长度为N的数列，数列中的每个数都属于集合S。

小C用这个生成器生成了许多这样的数列。但是小C有一个问题需要你的帮助：给定整数x，求所有可以生成出的，且满足数列中所有数的乘积mod M的值等于x的不同的数列的有多少个。小C认为，两个数列{Ai}和{Bi}不同，当且仅当至少存在一个整数i，满足Ai≠Bi。另外，小C认为这个问题的答案可能很大，因此他只需要你帮助他求出答案mod 1004535809的值就可以了。

1<=N<=10^9，3<=M<=8000，M为质数，0<=x<=M-1，输入数据保证集合S中元素不重复

 

首先因为M是质数并且0<=x和S中的元素<=M-1，就说明M有原根而且能很快求出来

于是对于i∈[0,M-1]的数都可以表示成原根的ki(k∈[0,M-1])次幂形式而且k互补相同

于是乘法就转化成了幂的加法，问题变成给|S|个数，求选n个数使得这n个数的和膜M==x的方案数（一个数可以选多次

所以生成函数，对于每个集合中的数si的ki，用ai表示kj==i有多少个（实际上最多只有一个，因为ki互不相同，同时S中的数互不相同

那么最终生成的多项式就是A=a_0+a_1*x+a_2*x^2+……a_{m-1}*x^{m-1}

因为有n个物品，所以多项式卷积n次

因为给的模数是恩梯梯模数，所以使用NTT计算精确答案

最后的答案就是(A^n)的第k_{x}相

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
#define ll long long
const ll mo=1004535809;
int rd(){int z=0,mk=1;  char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')mk=-1;  ch=getchar();}
    while(ch>='0'&&ch<='9'){z=(z<<3)+(z<<1)+ch-'0';  ch=getchar();}
    return z*mk;
}
int wtp=0,wtc[32];
void wt(int x,char y){
    if(!x){  putchar('0');  return ;}
    if(x<0)  putchar('-'),x=-x;
    while(x)  wtc[++wtp]=x%10+'0',x/=10;
    while(wtp)  putchar(wtc[wtp--]);
    putchar(y);
}
ll n,m,t,s;  int g,tg[210000];
ll a[210000],b[210000],tmp[210000],_x,_y;
int rvs[210000],dg[32],N,L;  ll _1_N;
int stck[210000],tp=0;
ll qcp(ll x,int y,int p){
    ll z=1,bs=x;
    for(;y;y>>=1){
        if(y&1)  z=(z*bs)%p;
        bs=(bs*bs)%p;
    }
    return z;
}
int gtg(){
    int _m=m-1;
    for(int i=2;i<=_m;++i)if(!(_m%i)){
        stck[++tp]=i;
        while(!(_m%i))  _m/=i;
    }
    for(int i=2,mk=false;;++i,mk=false){
        for(int j=1;j<=tp;++j)
            if(qcp(i,(m-1)/stck[j],m)==1){
                mk=true;
                break;
            }
        if(!mk)  return i;
    }
}
void ntt(ll x[],ll mk){
    for(int i=0;i<N;++i)  tmp[i]=x[rvs[i]];
    for(int i=0;i<N;++i)  x[i]=tmp[i];
    for(int i=2;i<=N;i<<=1){
        ll wn=qcp(3,(mk*((mo-1)/i))%(mo-1),mo);
        for(int k=0;k<N;k+=i){
            ll w=1;
            for(int j=k;j<k+(i>>1);++j){
                _x=x[j],_y=(x[j+(i>>1)]*w)%mo;
                x[j]=(_x+_y)%mo,x[j+(i>>1)]=(_x-_y+mo)%mo;
                w=(w*wn)%mo;
            }
        }
    }
    if(mk==mo-2)  for(int i=0;i<N;++i)  x[i]=(x[i]*_1_N)%mo;
}
void cclt(){
    b[0]=1;
    for(;n;n>>=1){
        ntt(a,1);
        if(n&1){
            ntt(b,1);  for(int i=0;i<N;++i)  b[i]=(b[i]*a[i])%mo;
            ntt(b,mo-2);  for(int i=N-1;i>=m-1;--i)  b[i-m+1]=(b[i-m+1]+b[i])%mo,b[i]=0;
        }
        for(int i=0;i<N;++i)  a[i]=(a[i]*a[i])%mo;
        ntt(a,mo-2);
        for(int i=N-1;i>=m-1;--i)  a[i-m+1]=(a[i-m+1]+a[i])%mo,a[i]=0;
    }
}
int main(){//freopen("ddd.in","r",stdin);
    cin>>n>>m>>t>>s;
    g=gtg();
    for(int i=0,x=1;i<m-1;++i,x=(x*g)%m)  tg[x]=i;
    int x;
    for(int i=1;i<=s;++i){
        x=rd();
        if(x)  ++a[tg[x]];
    }
    for(N=1,L=0;N<=m;N<<=1,++L);  N<<=1,++L;
    _1_N=qcp(N,mo-2,mo);
    for(int i=0;i<N;++i){
        for(int j=i,k=0;j;j>>=1,++k)  dg[k]=j&1;
        for(int j=0;j<L;++j)  rvs[i]=(rvs[i]<<1)|dg[j];
    }
    cclt();
    cout<<b[tg[t]]<<endl;
    return 0;
}