莫比乌斯反演入门(一)--P2522

前置知识:

1.数论分块,二维数论分块

2.$\lfloor \frac{a}{bc}\rfloor =\lfloor \frac{\lfloor \frac{a}{b}\rfloor }{c}\rfloor$

3.$gcd(i,j)=k\to gcd(\frac{i}{k},\frac{j}{k})=1$

OI WIKI 上面莫比乌斯反演前置知识

题目大意:

地址

求${\sum }_{i=a}^b{\sum }_{j=c}^d[gcd(i,j)==k]$

$所有变量\le 5e4$

题目思路:

莫比乌斯反演主要就是公式推导。

这个二维区间函数和容斥一下变成求${\sum }_{i=1}^n{\sum }_{j=1}^m[gcd(i,j)==k]$

原式 $={\sum }_{i=1}^n{\sum }_{j=1}^m[gcd(\frac{i}{k},\frac{j}{k})==1]$

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

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

接下来变换求和顺序。按约数来计算$\mu$的和

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

显然,$[1,\lfloor \frac{n}{k}\rfloor ]$内有$\lfloor \frac{\lfloor \frac{n}{k}\rfloor }{d}\rfloor$个$d$的倍数

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

线性筛筛出莫比乌斯函数,然后二维数论分块即可。

AC代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
#define vi vector<int>
#define vll vector<ll>
#define fi first
#define se second
const int maxn = 5e4 + 5;
const int mod = 1e9 + 7;
int a[maxn];
int u[maxn] , bk[maxn] , p[maxn] , cnt;
void init ()
{
    u[1] = 1;
    for (int i = 2 ; i < maxn ; i++){
        if (!bk[i]){
            p[++cnt] = i;
            u[i] = -1;
        }
        for (int j = 1 ; j <= cnt && i * p[j] < maxn ; j++){
            bk[i * p[j]] = 1;
            if (i % p[j] == 0) {
                u[i * p[j]] = 0;
                break;
            }else {
                u[i * p[j]] = -u[i];
            }
        }
    }
    for (int i = 1 ; i < maxn ; i++) u[i] += u[i - 1];
    return ;
}
ll solve (ll n , ll m)
{
    ll ans = 0;
    for (ll l = 1 , r ; l <= min(n , m) ; l = r + 1){
        r = min(n / (n / l) , m / (m / l));
        ans += (u[r] - u[l - 1]) * (n / l) * (m / l);
    }
    return ans;
}
int main()
{
    ios::sync_with_stdio(false);
    init();
    int t; cin >> t;
    while (t--){
        ll a , b , c , d , k;
        cin >> a >> b >> c >> d >> k;
        // (a , c) (b , d)
        ll ans = solve(b / k , d / k) - solve((a - 1) / k , d / k)
                - solve(b / k , (c - 1) / k) + solve((a - 1) / k , (c - 1) / k);
        cout << ans << endl;
    }
    return 0;
}