一道推起来很爽的组合优化题

sophilex

已于 2024-04-02 11:50:19 修改

阅读量907

点赞数 29

分类专栏：数学比赛题解文章标签：算法组合数学

于 2024-04-02 11:48:00 首次发布

本文链接：https://blog.csdn.net/sophilex/article/details/137264419

版权

比赛题解同时被 2 个专栏收录

33 篇文章 1 订阅

订阅专栏

数学

19 篇文章 0 订阅

订阅专栏

link
大意就是求 $\sum_{i=1}^{n}\sum_{j=1}^{m}\frac{(i,j)}{ij}$ ，其中 $n,m\leq 1e6,有1e4次查询$
原式子

$=\sum_{d=1}^{min(n,m)}\sum_{i=1}^{n}\sum_{j=1}^{m}\frac{d}{ij}[(i,j)=d]$

$=\sum_{d=1}^{min(n,m)}\sum_{i=1}^{n}\sum_{j=1}^{m}\frac{\frac{1}{d}}{\frac{i}{d}\frac{j}{d}}[(\frac{i}{d},\frac{j}{d})=1][d|i][d|j]$

$=\sum_{d=1}^{min(n,m)}\sum_{i=1}^{n}\sum_{j=1}^{m}\frac{\frac{1}{d}}{\frac{i}{d}\frac{j}{d}}[d|i][d|j]\sum_{t|(\frac{i}{d},\frac{j}{d})}\mu(t)$

$=\sum_{d=1}^{min(n,m)}\frac{1}{d}\sum_{t=1}^{\frac{min(n,m)}{d}}\mu(t)\sum_{i=1}^{n}\sum_{j=1}^{m}\frac{1}{\frac{i}{d}\frac{j}{d}}[dt|i][dt|j]$

$=\sum_{d=1}^{min(n,m)}\frac{1}{d}\sum_{t=1}^{\frac{min(n,m)}{d}}\mu(t)\sum_{i=1}^{\frac{n}{dt}}\sum_{j=1}^{\frac{m}{dt}}\frac{1}{\frac{i}{d}\frac{j}{d}d^2t^2}$

$=\sum_{d=1}^{min(n,m)}\frac{1}{d}\sum_{t=1}^{\frac{min(n,m)}{d}}\frac{1}{t^2}\mu(t)\sum_{i=1}^{\frac{n}{dt}}\sum_{j=1}^{\frac{m}{dt}}\frac{1}{ij}$

$=\sum_{d=1}^{min(n,m)}\frac{1}{d}\sum_{t=1}^{\frac{min(n,m)}{d}}\frac{1}{t^2}\mu(t)(\sum_{i=1}^{\frac{n}{dt}}\frac{1}{i})(\sum_{j=1}^{\frac{m}{dt}}\frac{1}{j})$
推到这里询问的复杂度来到了 $(O (n l o g))$ ,预处理倒数和之后直接暴力枚举即可。但是注意到有多次询问，所以还是要化到最简才可以

令 $f(n,D)=\sum_{i=1}^{\frac{n}{D}}\frac{1}{i},f(m,D)=\sum_{i=1}^{\frac{m}{D}}\frac{1}{j}$
原式

$=\sum_{D=1}^{min(n,m)}f(n,D)f(m,D)\sum_{d|D}\frac{1}{d}\frac{1}{(D/d)^2}\mu(\frac{D}{d})$

$=\sum_{D=1}^{min(n,m)}f(n,D)f(m,D)/D^2\sum_{d|D}d\mu(\frac{D}{d})$
注意到后面是一个欧拉反演

$=\sum_{D=1}^{min(n,m)}\frac{f(n,D)f(m,D)}{D^2}\phi(D)$

使用二阶整除分块的话，在 $O (n)$ 预处理的基础上可以做到 $O (s q r t)$ 的单次查询复杂度
code

ll f[N],phi[N],vis[N],p[N];
ll in[N];
ll cnt;
ll ksm(ll x,ll y)
{
    ll ans=1;
    while(y)
    {
        if(y&1) ans=ans*x%mod;
        x=x*x%mod;
        y>>=1;
    }
    return ans;
}
ll inv(ll x)
{
    return ksm(x,mod-2);
}
void init(ll n)
{
	phi[1]=1;
	for(int i=2;i<=n;++i)
	{
		if(!vis[i]) p[++cnt]=i,phi[i]=i-1;
		for(int j=1;(j<=cnt)&&(i*p[j]<=n);++j)
		{
			vis[i*p[j]]=1;
			if(i%p[j]==0)
			{
				phi[i*p[j]]=phi[i]*p[j];
				break;
			}
			phi[i*p[j]]=phi[i]*(p[j]-1);
		}
    }
    for(ll i=1;i<=n;++i) f[i]=phi[i]*inv(i*i%mod)%mod;
    for(int i=1;i<=n;++i) f[i]=(f[i]+f[i-1])%mod;
    for(ll i=1;i<=n;++i) in[i]=inv(i);
    for(int i=1;i<=n;++i) in[i]=(in[i]+in[i-1])%mod;
}
ll n,m;
void solve() 
{
    cin>>n>>m;
    ll ans=0;
    for(ll l=2,r;l<=min(n,m);l=r+1)
    {
        r=min(n,m);r=min(r,n/(n/l));r=min(r,m/(m/l));
        (ans+=in[n/l]*in[m/l]%mod*((f[r]-f[l-1])%mod+mod)%mod)%=mod;
    }
    (ans+=in[n]*in[m]%mod)%=mod;
    cout<<ans<<endl;
}