bzoj 4816 [Sdoi2017]數字表格——反演
阿新 • • 發佈:2018-12-13
題目:https://www.lydsy.com/JudgeOnline/problem.php?id=4816
\( ans=\prod\limits_{d=1}^{n}f[d]^{\sum\limits_{l=1}^{\frac{n}{d}}\left\lfloor\frac{n}{l*d}\right\rfloor*\left\lfloor\frac{m}{l*d}\right\rfloor} \)
\(=\prod\limits_{D=1}^{n}\prod\limits_{d|D}f[d]^{\mu(\frac{D}{d})*\left\lfloor\frac{n}{D}\right\rfloor*\left\lfloor\frac{m}{D}\right\rfloor} \)
令 \( g(D)=\prod\limits_{d|D}f(d)^{\mu(\frac{D}{d})} \) ,就能做了。預處理 g 不要 \( \sqrt{n} \) 列舉約數,而 n*ln(n) 列舉倍數。
預處理 g 的字首積的逆元,回答詢問的時候就少一個 log 。
注意指數上是模 (mod-1) !!!
#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const int N=1e6+5,mod=1e9+7; int g[N],s[N],sn[N],u[N],pri[N];bool vis[N]; int pw(int x,int k) {int ret=1;while(k){if(k&1)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=1;}return ret;} void init() { int f0=0,f1=1,lm=1e6,cnt=0; u[1]=1; for(int i=2;i<=lm;i++) { if(!vis[i])pri[++cnt]=i,u[i]=-1; for(int j=1;j<=cnt&&(ll)i*pri[j]<=lm;j++) { vis[i*pri[j]]=1; if(i%pri[j]==0){u[i*pri[j]]=0;break;} u[i*pri[j]]=-u[i]; } } for(int j=1;j<=lm;j++)g[j]=1; s[1]=1; for(int i=2;i<=lm;i++) { swap(f0,f1);f1+=f0;if(f1>=mod)f1-=mod; int inv=pw(f1,mod-2); for(int j=i,k=1;j<=lm;j+=i,k++) if(u[k]==1)g[j]=(ll)g[j]*f1%mod; else if(u[k]==-1)g[j]=(ll)g[j]*inv%mod; s[i]=(ll)s[i-1]*g[i]%mod; } sn[lm]=pw(s[lm],mod-2); for(int i=lm-1;i;i--)sn[i]=(ll)sn[i+1]*g[i+1]%mod; sn[0]=1;// } int main() { int T,n,m;scanf("%d",&T);init(); while(T--) { scanf("%d%d",&n,&m);if(n>m)swap(n,m); int ans=1; for(int i=1,j;i<=n;i=j+1) { int d0=n/i,d1=m/i; j=min(n/d0,m/d1); ans=(ll)ans*pw((ll)s[j]*sn[i-1]%mod,(ll)d0*d1%(mod-1))%mod;/////%(mod-1)!!!!! } printf("%d\n",ans); } return 0; }