LOJ #528. 「LibreOJ β Round #4」求和 （莫比乌斯函数）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/sunshiness_s/article/details/81363403

题意

计算 $\sum\limits_{i=1}^n\sum\limits_{j=1}^m\mu^2(gcd(i,j))$ $(mod$ $998244353 )$

---

题解

$\sum\limits_{d=1}^n\mu^2(d)\sum\limits_{i=1}^{\frac{n}{d}}\sum\limits_{j=1}^{\frac{m}{d}}[gcd(i,j)==1]$

$\mu^2(x)$ 也就是只有 $x$ 不含平方数时才得 $1$

因此，我们只要找到有多少 $gcd$ 为平方数的就可以啦

---

代码

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define N 4000010
#define mod 998244353
#define ll long long
ll n,m,ans=0;
int tot,notprime[N],prime[N],mu[N];
inline void mobius(){
    mu[1]=1;
    for(int i=2;i<=4000000;i++){
        if(!notprime[i]) prime[++tot]=i,mu[i]=-1;;
        for(int j=1;j<=tot && prime[j]*i<=4000000;j++){
            notprime[prime[j]*i]=1;
            if(i%prime[j]==0){
                mu[prime[j]*i]=0;
                break;
            }mu[prime[j]*i]=-mu[i];
        }
    }
}
int main(){
    freopen("b.in","r",stdin);
    scanf("%lld%lld",&n,&m);
    if(n>m) swap(n,m);mobius();
    for(int i=1;(ll)i*i<=n;i++){
        ll j=(ll)i*i;if(!mu[i]) continue;
        ans=((ans+mu[i]*((n/j)%mod)%mod*((m/j)%mod)%mod)%mod+mod)%mod;
    }
    printf("%lld",(ans+mod)%mod);
    return 0;
}