P2522 [HAOI2011]Problem b (莫比乌斯反演)

时间:2023-11-30 19:02:20

题目

P2522 [HAOI2011]Problem b

解析:

具体推导过程同P3455 [POI2007]ZAP-Queries

不同的是,这个题求的是\(\sum_{i=a}^b\sum_{j=c}^dgcd(i,j)=k\)

像二维前缀和一样容斥一下,输出就完了。

根据luogu某大佬的说法

P2522 [HAOI2011]Problem b (莫比乌斯反演)

开longlong的话会TLE。。

代码

//莫比乌斯反演
#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
int t, n, m, num, k;
int mu[N], p[N], sum[N];
bool vis[N]; template<class T>inline void read(T &x) {
x = 0; int f = 0; char ch = getchar();
while (!isdigit(ch)) f |= (ch == '-'), ch = getchar();
while (isdigit(ch)) x = x * 10 + ch - '0', ch = getchar();
x = f ? -x : x;
return;
} void get_mu(int n) {
mu[1] = 1;
for (int i = 2; i <= n; ++i) {
if (!vis[i]) p[++num] = i, mu[i] = -1;
for (int j = 1; j <= num; ++j) {
if (i * p[j] > n) break;
vis[i * p[j]] = 1;
if (i % p[j] == 0) {
mu[i * p[j]] = 0;
break;
} else mu[i * p[j]] = -mu[i];
}
}
} int cal(int x, int y, int k) {
int mx = min(x, y), ans = 0;
for (int l = 1, r; l <= mx; l = r + 1) {
r = min(x / (x / l), y / (y / l));
ans += ((x / (l * k)) * (y / (l * k)) * (sum[r] - sum[l - 1]));
}
return ans;
} int a, b, c, d; signed main() {
get_mu(N);
for (int i = 1; i <= N; ++i) sum[i] = sum[i - 1] + mu[i];
read(t);
while (t --) {
read(a), read(b), read(c), read(d), read(k);
printf("%d\n", cal(b, d, k) - cal(b, c - 1, k) - cal(a - 1, d, k) + cal(a - 1, c - 1, k));
}
return 0;
}