https://www.lydsy.com/JudgeOnline/problem.php?id=3992
https://www.luogu.org/problemnew/show/P3321
小C有一个集合S,里面的元素都是小于M的非负整数。他用程序编写了一个数列生成器,可以生成一个长度为N的数列,数列中的每个数都属于集合S。小C用这个生成器生成了许多这样的数列。但是小C有一个问题需要你的帮助:给定整数x,求所有可以生成出的,且满足数列中所有数的乘积mod M的值等于x的不同的数列的有多少个。小C认为,两个数列{Ai}和{Bi}不同,当且仅当至少存在一个整数i,满足Ai≠Bi。另外,小C认为这个问题的答案可能很大,因此他只需要你帮助他求出答案mod 1004535809的值就可以了。
参考:http://www.cnblogs.com/candy99/p/6391483.html
打眼一看模数就是一个原根为3的MTT质数,先敲(抄)为敬。
然后求方案,美滋滋生成函数求完之后n次幂。
emmm然而生成函数只能做加法不能乘啊怎么办?
我们可以把乘法换成加法啊!
曾经做过一道题,把所有数取log……emmm这题显然不行。
(瞄了眼题解……
哇用原根表示数,这样乘法就表示成了指数之间的加法真是非常妙啊!
(当然注意对指数要对m-1取模才行)
#include<cstdio>
#include<cctype>
#include<cstring>
#include<vector>
#include<cmath>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long ll;
const ll P=;
const int G=;
const int N=3e4+;
inline int read(){
int X=,w=;char ch=;
while(!isdigit(ch)){w|=ch=='-';ch=getchar();}
while(isdigit(ch))X=(X<<)+(X<<)+(ch^),ch=getchar();
return w?-X:X;
}
ll qpow(ll a,ll n,ll p){
ll res=;
while(n){
if(n&)res=res*a%p;
a=a*a%p;n>>=;
}
return res;
}
void MTT(ll a[],int n,int on){
for(int i=,j=n>>;i<n-;i++){
if(i<j)swap(a[i],a[j]);
int k=n>>;
while(j>=k){j-=k;k>>=;}
if(j<k)j+=k;
}
for(int i=;i<=n;i<<=){
ll res=qpow(G,(P-)/i,P);
for(int j=;j<n;j+=i){
ll w=;
for(int k=j;k<j+i/;k++){
ll u=a[k],t=w*a[k+i/]%P;
a[k]=(u+t)%P;
a[k+i/]=(u-t+P)%P;
w=w*res%P;
}
}
}
if(on==-){
ll inv=qpow(n,P-,P);
a[]=a[]*inv%P;
for(int i=;i<=n/;i++){
a[i]=a[i]*inv%P;
if(i!=n-i)a[n-i]=a[n-i]*inv%P;
swap(a[i],a[n-i]);
}
}
}
vector<int>v;
bool pan(int g,int p){
for(int i=;i<v.size();i++){
if(qpow(g,(p-)/v[i],p)==)return ;
}
return ;
}
int primitive(int p){
int res=p-;v.clear();
for(int i=;i*i<=res;i++){
if(res%i==){
v.push_back(i);
while(res%i==)res/=i;
}
}
if(res!=)v.push_back(res);
for(int i=;;i++){
if(pan(i,p))return i;
}
}
int nn,n,m,x,s,sum,ind[N];
void multi(ll a[],ll b[]){
static ll c[N];
for(int i=;i<nn;i++)c[i]=b[i];
MTT(a,nn,);MTT(c,nn,);
for(int i=;i<nn;i++)a[i]=a[i]*c[i]%P;
MTT(a,nn,-);
for(int i=;i<m-;i++)
a[i]=(a[i]+a[i+m-])%P,a[i+m-]=;
}
void sqr(ll a[]){
MTT(a,nn,);
for(int i=;i<nn;i++)a[i]=a[i]*a[i]%P;
MTT(a,nn,-);
for(int i=;i<m-;i++)
a[i]=(a[i]+a[i+m-])%P,a[i+m-]=;
}
ll a[N],ans[N];
int main(){
n=read(),m=read(),x=read(),s=read();
int g=primitive(m),k=;
for(int i=;i<m-;i++){
ind[k]=i;k=k*g%m;
}
for(int i=;i<=s;i++){
int w=read();
if(w)a[ind[w]]=;
}
nn=;ans[]=;
while(nn<*m)nn<<=;
while(n){
if(n&)multi(ans,a);
sqr(a);n>>=;
}
printf("%lld\n",ans[ind[x]]);
return ;
}
+++++++++++++++++++++++++++++++++++++++++++
+本文作者:luyouqi233。 +
+欢迎访问我的博客:http://www.cnblogs.com/luyouqi233/ +
+++++++++++++++++++++++++++++++++++++++++++