[POI 2007] ZAP-Queries（莫比乌斯反演） | 错题本

题目

[POI2007]ZAP-Queries

分析

同 [HAOI 2011] Problem b。

代码

很迷惑的是筛莫比乌斯函数的时候要把 MAXN 开大一点点不然只有 80 pts。

#include <bits/stdc++.h>

typedef long long LL;

const int MAXN = 60000;

int Mu[MAXN + 5];
bool Vis[MAXN + 5];
std::vector<int> Primes;

void Init(int n) {
	Mu[1] = 1;
	for (int i = 2; i <= n; i++) {
		if (!Vis[i])
			Mu[i] = -1, Primes.push_back(i);
		for (int j = 0; j < (int)Primes.size() && i * Primes[j] < n; j++) {
			Vis[i * Primes[j]] = true;
			if (i % Primes[j] == 0) {
				Mu[i * Primes[j]] = 0;
				break;
			}
			Mu[i * Primes[j]] = -Mu[i];
		}
        Mu[i] += Mu[i - 1];
	}
}

LL Cal(int N, int M) {
    if (N < M) std::swap(N, M); LL Ans = 0;
    for (int lft = 1, rgt = 1; lft <= M; lft = rgt + 1) {
        rgt = std::min(M / (M / lft), N / (N / lft));
        Ans += (LL)(N / lft) * (M / lft) * (Mu[rgt] - Mu[lft - 1]);
    }
    return Ans;
}

int main() {
    Init(MAXN);
    int T; scanf("%d", &T);
    while (T--) {
        int A, B, K; scanf("%d%d%d", &A, &B, &K);
        printf("%lld\n", Cal(A / K, B / K));
    }
    return 0;
}