1. 程式人生 > >bzoj 4816 [Sdoi2017]數字表格——反演

bzoj 4816 [Sdoi2017]數字表格——反演

題目: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; }