【BZOJ 2301】problem b

【题目描述】

对于给出的$n$个询问，每次求有多少个数对$(x,y)$，满足$a\le x\le b，c\le y\le d$，且$gcd(x,y)=k$.

【分析】

记
$$f(a,b)=\sum _{i=1}^a\sum _{j=1}^b[gcd(i,j)=k]$$
则答案可以由上式差分而来。故我们只需考虑如何计算它。
枚举k的倍数：
$$\sum _{i=1}^{\lfloor \frac{a}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{b}{k}\rfloor }[gcd(i,j)=1]$$
由${\sum }_{d|n}\mu (d)=[n=1]$：
$$\sum _{i=1}^{\lfloor \frac{a}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{b}{k}\rfloor }\sum _{d|gcd(i,j)}\mu (d)$$
这样的话，实际我们只是在研究对于每个d，$\mu (d)$被统计了多少次。
这样问题就变得简单了：我只需统计对于每个d，有多少个i和j同时是d的倍数即可
而我们知道，在$[1,n]$范围内，数d的倍数的个数$=\lfloor n/d\rfloor$。
所以：
$$\sum _{d=1}^{\lfloor \frac{max(a,b)}{k}\rfloor }\lfloor \frac{a}{kd}\rfloor \lfloor \frac{b}{kd}\rfloor$$
数论分块即可。

【代码】

#include<bits/stdc++.h>
#pragma GCC optimize (3)
#define ll long long
using namespace std;
const int mn = 50005;
int mu[mn];
bool vis[mn];
int pri[mn], cnt, k;
inline void init(int n)
{
    mu[1] = 1;
    for(int i = 2; i <= n; i++)
    {
        if(!vis[i]) pri[++cnt] = i, mu[i] = -1;
        for(int j = 1; j <= cnt && pri[j] * i <= n; j++)
        {
            vis[i * pri[j]] = 1;
            if(i % pri[j] == 0) {mu[pri[j] * i] = 0; break;}
            else mu[pri[j] * i] = -mu[i];
        }
    }
    for(int i = 2; i <= n; i++)
        mu[i] += mu[i - 1];
}
inline ll calc(int x, int y)
{
    x /= k, y /= k;
    if(x > y) swap(x, y);
    ll ret = 0;
    for(int i = 1, las; i <= x; i = las + 1)
    {
        las = min(x / (x / i), y / (y / i));
        ret += 1ll * (mu[las] - mu[i - 1]) * (x / i) * (y / i);
    }
    return ret;
}
int main()
{
    int a, b, c, d, T;
    scanf("%d", &T), init(mn - 5);
    while(T--)
    {
        scanf("%d%d%d%d%d", &a, &b, &c, &d, &k);
        printf("%lld\n", calc(b, d) - calc(a-1, d) - calc(b, c - 1) + calc(a-1, c-1));
    }
}