題目鏈接：點(diǎn)擊查看

題目大意：給出一個集合 $S$，集合中的數(shù)是 $[0,m)$ 且互不相同的，問從集合中選 $n$ 次數(shù)字，且乘積對 $m$ 取模后等于 $x$ 的方案數(shù)有多少

題目分析：

考慮轉(zhuǎn)移方程，我們設(shè) $f[i][j]$ 為選了 $i$ 個數(shù)后，乘積對 $m$ 取余后為 $j$ 的方案數(shù)，那么自然有 $f[i+1][j]=\sum _{ab\equiv j(\mathrm{m}\mathrm{o}\mathrm{d}m)}f[i][a]?f[i][b]$

需要注意到本題的 $m$ 是一個特別小的質(zhì)數(shù)，不難想到原根，設(shè) $g$ 為 $m$ 的原根，可以得到一個性質(zhì)就是 $g^1,g^2,...,g^{m?1}$ 各不相同，且值域?qū)儆?$[1,m)$

所以將每個數(shù)換成其原根對應(yīng)的數(shù)字后就可以化乘為加了，設(shè) $g^A\equiv a(\mathrm{m}\mathrm{o}\mathrm{d}m)$ 更具體的，原轉(zhuǎn)移方程就可以書寫為： $$f[i+1][j]=\sum _{g^A{g}^B\equiv g^J(\mathrm{m}\mathrm{o}\mathrm{d}m)}f[i][A]?f[i][B]$$ 因?yàn)?$m$ 是質(zhì)數(shù)，所以根據(jù)費(fèi)馬小定理降冪得到：$$f[i+1][j]=\sum _{g^{A+B(\mathrm{m}\mathrm{o}\mathrm{d}m?1)}\equiv g^J(\mathrm{m}\mathrm{o}\mathrm{d}m)}f[i][A]?f[i][B]$$

即 $$f[i+1][j]=\sum _{A+B\equiv J(\mathrm{m}\mathrm{o}\mathrm{d}m?1)}f[i][A]?f[i][B]$$

又因?yàn)槊繉拥木矸e的轉(zhuǎn)移方程是相同的，可以用快速冪優(yōu)化NTT，時間復(fù)雜度為 $O(m?logm?logn)$

有幾個小細(xì)節(jié)需要注意：

因?yàn)槟?shù)從 $m$ 變成了 $m?1$，原本 $g^{m?1}\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}m)$，這里的指數(shù)需要對 $m?1$ 取模，換算到本題中應(yīng)該變成 $g^{m?1}=g^0=1$，所以當(dāng)原數(shù)為 $1$ 時，對應(yīng)的原根的指數(shù)應(yīng)該為 $0$，幾乎沒有題解說明這個細(xì)節(jié)，自己想了好久才明白的

$S$ 集合中的 $0$ 是沒有貢獻(xiàn)的，直接忽略即可

每次卷完之后，因?yàn)橄聵?biāo)的取值是 $[0,2?(m?1))$ 的，所以需要將 $[m?1,2?(m?1))$ 這一部分的答案加到 $[0,m?1)$ 上去

代碼：

// Problem: P3321 [SDOI2015]序列統(tǒng)計 // Contest: Luogu // URL: https://www.luogu.com.cn/problem/P3321 // Memory Limit: 125 MB // Time Limit: 1000 ms // // Powered by CP Editor (https://cpeditor.org)// #pragma GCC optimize(2) // #pragma GCC optimize("Ofast","inline","-ffast-math") // #pragma GCC target("avx,sse2,sse3,sse4,mmx") #include<iostream> #include<cstdio> #include<string> #include<ctime> #include<cmath> #include<cstring> #include<algorithm> #include<stack> #include<climits> #include<queue> #include<map> #include<set> #include<sstream> #include<cassert> #include<bitset> #include<list> #include<unordered_map> #define lowbit(x) (x&-x) using namespace std; typedef long long LL; typedef unsigned long long ull; template<typename T> inline void read(T &x) {T f=1;x=0;char ch=getchar();while(0==isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}while(0!=isdigit(ch)) x=(x<<1)+(x<<3)+ch-'0',ch=getchar();x*=f; } template<typename T> inline void write(T x) {if(x<0){x=~(x-1);putchar('-');}if(x>9)write(x/10);putchar(x%10+'0'); } const int inf=0x3f3f3f3f; const int N=1e6+100; const int mod=1004535809,G=3,Gi=(mod+1)/3; bool vis[8010]; int n,m,limit = 1,L,r[N],p[8010]; LL a[N],ans[N]; inline LL fastpow(LL a, LL k) {LL base = 1;while(k) {if(k & 1) base = (base * a ) % mod;a = (a * a) % mod;k >>= 1;}return base % mod; } inline void NTT(LL *A, int type) {for(int i = 0; i < limit; i++) if(i < r[i]) swap(A[i], A[r[i]]);for(int mid = 1; mid < limit; mid <<= 1) { LL Wn = fastpow( type == 1 ? G : Gi , (mod - 1) / (mid << 1));for(int j = 0; j < limit; j += (mid << 1)) {LL w = 1;for(int k = 0; k < mid; k++, w = (w * Wn) % mod) {int x = A[j + k], y = w * A[j + k + mid] % mod;A[j + k] = (x + y) % mod,A[j + k + mid] = (x - y + mod) % mod;}}}if(type==-1) {LL inv=fastpow(limit,mod-2);for(int i=0;i<limit;i++) {A[i]=(A[i]*inv)%mod;}} } void q_pow(int n) {ans[0]=1;while(n) {NTT(a,1);if(n&1) {NTT(ans,1);for(int i=0;i<limit;i++) {ans[i]=ans[i]*a[i]%mod;}NTT(ans,-1);for(int i=limit-1;i>=m-1;i--) {ans[i-m+1]=(ans[i-m+1]+ans[i])%mod;ans[i]=0;}}for(int i=0;i<limit;i++) {a[i]=a[i]*a[i]%mod;}NTT(a,-1);for(int i=limit-1;i>=m-1;i--) {a[i-m+1]=(a[i-m+1]+a[i])%mod;a[i]=0;}n>>=1;} } int find_root(int mod) {for(int i=2;i<mod;i++){memset(vis,false,sizeof(vis));bool flag=true;int x=i;for(int j=1;j<=mod-1;j++){if(vis[x]){flag=false;break;}vis[x]=true;x=x*i%mod;}if(flag)return i;}return -1; } int main() { #ifndef ONLINE_JUDGE // freopen("data.in.txt","r",stdin); // freopen("data.out.txt","w",stdout); #endif // ios::sync_with_stdio(false);int x,S;read(n),read(m),read(x),read(S);while(limit<=m+m) limit<<=1,L++;for(int i=0;i<limit;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(L-1));int rt=find_root(m);int sum=1;for(int i=1;i<m-1;i++) {sum=sum*rt%m;p[sum]=i;}for(int i=0,x;i<S;i++) {read(x);if(x) {a[p[x]]=1;}}q_pow(n);cout<<ans[p[x]]<<endl;return 0; } 超強(qiáng)干貨來襲 云風(fēng)專訪：近40年碼齡，通宵達(dá)旦的技術(shù)人生

總結(jié)

以上是生活随笔為你收集整理的洛谷 - P3321 [SDOI2015]序列统计(原根+NTT)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。