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

题目

P2522 [HAOI2011]Problem b

解析：

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

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

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

根据luogu某大佬的说法



开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;

}