Luogu P3455 [POI2007]ZAP-Queries

题目大意

给定 $n,m,k$，求
$$\sum _{i=1}^n\sum _{j=1}^m[(i,j)=k]$$

题解

考虑化简，得：
$$\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{m}{k}\rfloor }[(i,j)=1]$$
利用中心结论，得：
$$\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{m}{k}\rfloor }\sum _{d|(i,j)}\mu (d)$$

$$\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{m}{k}\rfloor }\sum _{d=1}^{(i,j)}\mu (d)[d|(i,j)]$$

$$\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{m}{k}\rfloor }\sum _{d=1}^{(i,j)}\mu (d)[d|i][d|j]$$

$\because gcd(a,b)⩽min(a,b)$
设答案为

$$\sum _{d=1}^{min(\lfloor \frac{n}{k}\rfloor ,\lfloor \frac{m}{k}\rfloor )}{p}_d\ast \mu (d)。$$

我们发现，$p_d$ 等于 $d|i$ 且 $d|j$ 的数的个数。
因此可以转换成如下形式：
$$\sum _{d=1}^{min(\lfloor \frac{n}{k}\rfloor ,\lfloor \frac{m}{k}\rfloor )}\mu (d)\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }[d|i]\sum _{j=1}^{\lfloor \frac{m}{k}\rfloor }[d|j]$$
后面两个和式可以直接得出：
$$\sum _{d=1}^{min(\lfloor \frac{n}{k}\rfloor ,\lfloor \frac{m}{k}\rfloor )}\mu (d)\lfloor \frac{n}{kd}\rfloor \lfloor \frac{m}{kd}\rfloor$$
线筛出 $\mu$，求前缀和，最后数论分块即可。时间复杂度 $\Theta (n+T\sqrt{n})$。

#include<bits/stdc++.h>
using namespace std;
#define maxn 100000
int mu[maxn],p[maxn];
bool vis[maxn];
void euler_mu(int n) {
    int tot=0;
    memset(mu,0,sizeof mu);
    memset(vis,0,sizeof vis);
    mu[1]=1;
    for(int i=2;i<=n;++i) {
        if(!vis[i]) p[++tot]=i,mu[i]=-1;
        for(int j=1;j<=tot&&i*p[j]<=n;++j) {
            vis[i*p[j]]=1;
            if(i%p[j]==0) {
                mu[i*p[j]]=0;
                break;
            }
            mu[i*p[j]]=-mu[i];
        }
    } for(int i=1;i<=n;i++)//前缀和
        mu[i]+=mu[i-1];
}
int main(){
    int T;
    euler_mu(50000);//线性筛只做一次
    scanf("%d",&T);
    while(T--){
        int _n,_m,k,ans=0;
        scanf("%d%d%d",&_n,&_m,&k);
        _n/=k;_m/=k;
        int a=_n,b=_m,n=min(_n,_m);
        int len=min(n,min(a,b));
        for(int l=1,r;l<=len;l=r+1){
            r=min(a/(a/l),b/(b/l));
            if(r>n) r=n;
            ans+=(mu[r]-mu[l-1])*(a/l)*(b/l);
        }
        printf("%d\n",ans);
    }
    return 0;
}