Luogu P3455 [POI2007]ZAP-Queries

题目描述

https://www.luogu.com.cn/problem/P3455

简要题意：求 $\sum_{i=1}^n\sum_{j=1}^m[(i,j)=k]$

$n,m,k\le 5\times 10^4$

Solution

$$\begin{aligned} ans&=\sum_{i=1}^n\sum_{j=1}^m[(i,j)=k]\\ &=\sum_{i=1}^n\sum_{j=1}^m[(\frac{i}{k},\frac{j}{k})=1]\\ &=\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_{d=1}^{\lfloor\frac{n}{k}\rfloor}\mu(d)\sum_{d|i}\sum_{d|j}1\\ &=\sum_{d=1}^{\lfloor\frac{n}{k}\rfloor}\mu(d)\lfloor\frac{n}{dk}\rfloor\lfloor\frac{m}{dk}\rfloor\\ \end{aligned}$$

#include <iostream>
#include <cstdio>
#define maxn 50010
#define ll long long
using namespace std;

int n, m, k; 

int pri[maxn], cnt, mu[maxn]; bool isp[maxn];
void init_isp(int n) {
    mu[1] = 1;
    for (int i = 2; i <= n; ++i) {
        if (!isp[i]) pri[++cnt] = i, mu[i] = -1;
        for (int j = 1; j <= cnt && i * pri[j] <= n; ++j) {
            isp[i * pri[j]] = 1;
            if (i % pri[j] == 0) break;
            mu[i * pri[j]] = -mu[i]; 
        }
    }
    for (int i = 1; i <= n; ++i) mu[i] += mu[i - 1];
}

void work() {
    cin >> n >> m >> k; n /= k; m /= k; 
    if (n > m) swap(n, m); ll ans = 0; 
    for (ll l = 1, r; l <= n; l = r + 1) {
        r = min(n / (n / l), m / (m / l));
        ans += (mu[r] - mu[l - 1]) * (n / l) * (m / l);
    }
    cout << ans << "\n"; 
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr); cout.tie(nullptr);

    int T; cin >> T; init_isp(50000); 
    while (T--) work(); 
    return 0;
}