BZOJ.4816.[SDOI2017]数字表格(莫比乌斯反演)

时间:2023-02-21 19:33:13

题目链接

总感觉博客园的\(Markdown\)很。。\(gouzhi\),可以看这的

这个好像简单些啊,只要不犯sb错误

**[Update] **真的算反演中比较裸的题了...


\(Description\)

  用\(f[i]\)表示\(Fibonacci\)数列的第\(i\)项,求$$\prod_{i=1}n\prod_{j=1}mf[\gcd(i,j)]\mod (10^9+7)$$

\(Solution\)

\[ \begin{aligned}
Ans &=\prod_{i=1}^n\prod_{j=1}^mf[(i,j)]\\
&=\prod_{d=1}^n\prod_{i=1}^n\prod_{j=1}^m[(i,j)=d]*f[d]\\
&=\prod_{d=1}^nf[d]^{\sum_{i=1}^n\sum_{j=1}^m[(i,j)=d]}\\
&=\prod_{d=1}^nf[d]^{\sum_{i=1}^{\lfloor\frac{n}{d}\rfloor}\sum_{j=1}^{\lfloor\frac{m}{d}\rfloor}[(i,j)=1]}\\
&=\prod_{d=1}^nf[d]^{\sum_{i=1}^{min}\mu(i)\lfloor\frac{n}{id}\rfloor\lfloor\frac{m}{id}\rfloor}\\
&=\prod_{d=1}^nf[d]^{\sum_{d\mid T}\mu(\frac{T}{d})\lfloor\frac{n}{T}\rfloor\lfloor\frac{m}{T}\rfloor}
\end{aligned}
\]

  直接去枚举\(T\),

\[ \begin{aligned}
Ans &=\prod_{d=1}^nf[d]^{\sum_{d\mid T}\mu(\frac{T}{d})\lfloor\frac{n}{T}\rfloor\lfloor\frac{m}{T}\rfloor}\\
&=\prod_{T=1}^n\prod_{d\mid T}f[d]^{\mu(\frac{T}{d})\lfloor\frac{n}{T}\rfloor\lfloor\frac{m}{T}\rfloor}\\
&=\prod_{T=1}^n\prod_{d\mid T}(f[d]^{\mu(\frac{T}{d}) })^{\lfloor\frac{n}{T}\rfloor\lfloor\frac{m}{T}\rfloor}
\end{aligned}
\]

  同之前,枚举约数暴力更新\(\prod_{d\mid T}f[d]^{\mu(\frac{T}{d})}\),复杂度\(O(n\log n)\)(\(\prod_{d\mid T}f[\frac{T}{d}]^{\mu(d)}\))。总复杂度\(O(n\log n+T\sqrt n)\)。

  注意次幂不能直接取模,利用费马小定理可以对\(mod-1\)取模。

  然后注意\(f[0]=0\)。。

小结:

  套路1:把\(\prod\)换到幂上去,这样就有\(\sum\)了。

  套路2:令\(T=id\),在最外层枚举\(T\)。

  另外常见的就不再写了。

  F[]用int,FP()中的长式子放到外面,优化还是比较明显有的。


原先的代码:

//17912kb	35060ms
#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
#define mod (1000000007)
typedef long long LL;
const int N=1e6+2; int cnt,P[N>>3],mu[N+2],f[N+2],inv_f[N+2],F[N+2];
bool Not_p[N+2]; inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-'0',c=gc());
return now;
}
int FP(LL x,int k)
{
LL t=1;
for(; k; k>>=1,x=x*x%mod)
if(k&1) t=t*x%mod;
return t;
}
void Init()
{
mu[1]=f[1]=inv_f[1]=1;//f[0]=0!
for(int i=2; i<N; ++i)
f[i]=(f[i-1]+f[i-2])%mod, inv_f[i]=FP(f[i],mod-2);
for(int i=2; i<N; ++i)
{
if(!Not_p[i]) P[++cnt]=i,mu[i]=-1;
for(int v,j=1; j<=cnt&&(v=i*P[j])<N; ++j)
{
Not_p[v]=1;
if(i%P[j]) mu[v]=-mu[i];
else {mu[v]=0; break;}
}
}
for(int i=0; i<N; ++i) F[i]=1;//F[0]=1
for(int i=1; i<N; ++i)
if(mu[i]==1)
for(int d=1,T=0; (T+=i)<N; ++d) F[T]=1ll*F[T]*f[d]%mod;
else if(mu[i]==-1)
for(int d=1,T=0; (T+=i)<N; ++d) F[T]=1ll*F[T]*inv_f[d]%mod;
for(int i=2; i<N; ++i) F[i]=1ll*F[i]*F[i-1]%mod;
}
inline int inv(int x){
return FP(x,mod-2);
}
int Calc(int n,int m)
{
if(n>m) std::swap(n,m);
int res=1;
for(int nxt,tmp,i=1; i<=n; i=nxt+1){
nxt=std::min(n/(n/i),m/(m/i)),
tmp=1ll*F[nxt]*inv(F[i-1])%mod,//存次数就没用啦?大概是留一个式子比较好?
res=1ll*res*FP(tmp,1ll*(n/i)*(m/i)%(mod-1))%mod;//注意次幂是模(mod-1)。
}
return res;
} int main()
{
Init();
int T,n,m; scanf("%d",&T);
while(T--)
scanf("%d%d",&n,&m),printf("%d\n",Calc(n,m)); return 0;
}

新写的代码(19.2.13):

/*
17912kb 33888ms
注意本题特殊在是次幂=-=:mu[]可能等于-1,即乘一个F[]的逆元;指数要对mod-1取模。
*/
#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
#define mod 1000000007
#define Mod(x) x>=mod&&(x-=mod)
typedef long long LL;
const int N=1e6+2; int sum[N],F[N]; inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-48,c=gc());
return now;
}
inline int FP(int x,int k)
{
int t=1;
for(; k; k>>=1,x=1ll*x*x%mod)
if(k&1) t=1ll*t*x%mod;
return t;
}
void Init()
{
static int cnt,P[N>>3],mu[N],invF[N];
static bool notP[N];
F[0]=0, F[1]=F[2]=invF[1]=invF[2]=1;
for(int i=3; i<N; ++i) F[i]=F[i-1]+F[i-2], Mod(F[i]), invF[i]=FP(F[i],mod-2);
mu[1]=1;
for(int i=2; i<N; ++i)
{
if(!notP[i]) P[++cnt]=i, mu[i]=-1;
for(int j=1,v; j<=cnt&&(v=i*P[j])<N; ++j)
{
notP[v]=1;
if(i%P[j]) mu[v]=-mu[i];
else {mu[v]=0; break;}
}
}
for(int i=0; i<N; ++i) sum[i]=1;//sum[0]=1,因为询问的时候要用到*inv(sum[0]) = =
for(int d=1; d<N; ++d)//可以先枚举这个mu[d],就可以在循环外写if了=v=
if(mu[d]==1)
for(int T=0,i=1; (T+=d)<N; ++i) sum[T]=1ll*sum[T]*F[i]%mod;
else if(mu[d]==-1)
for(int T=0,i=1; (T+=d)<N; ++i) sum[T]=1ll*sum[T]*invF[i]%mod;
for(int i=2; i<N; ++i) sum[i]=1ll*sum[i]*sum[i-1]%mod;
} int main()
{
Init();
for(int T=read(); T--; )
{
const int n=read(),m=read();
LL ans=1;
for(int i=1,l=std::min(n,m),nxt; i<=l; i=nxt+1)
{
nxt=std::min(n/(n/i),m/(m/i));
ans=ans*FP(1ll*sum[nxt]*FP(sum[i-1],mod-2)%mod,1ll*(n/i)*(m/i)%(mod-1))%mod;
}
printf("%lld\n",ans);
} return 0;
}