[洛谷 P2257] YY的GCD（莫比乌斯反演） | 错题本

最新推荐文章于 2020-11-30 20:33:31 发布

ixRic

最新推荐文章于 2020-11-30 20:33:31 发布

阅读量199

点赞数 1

分类专栏： [ 错题本 ] # 莫比乌斯反演文章标签：莫比乌斯函数

本文链接：https://blog.csdn.net/C20190102/article/details/108416374

版权

[ 错题本 ] 同时被 2 个专栏收录

86 篇文章 1 订阅

订阅专栏

莫比乌斯反演

13 篇文章 0 订阅

订阅专栏

文章目录

题目
分析
代码

题目

[洛谷 P2257] YY的GCD

分析

不妨设 $\geq M$ ， $P$ 是素数集，枚举 $\gcd(x, y), i = \frac{x}{d}, j = \frac{y}{d}$ ，要求的即 $\begin{aligned} &\sum_{d = 1}^{M} [d \in P]\sum_{i = 1}^{\left\lfloor\frac{N}{d}\right\rfloor} \sum_{j = 1}^{\left\lfloor\frac{M}{d}\right\rfloor} [\gcd(i, j) = 1] \\ =&\sum_{d = 1}^{M} [d \in P] \sum_{i = 1}^{\left\lfloor\frac{N}{d}\right\rfloor} \sum_{j = 1}^{\left\lfloor\frac{M}{d}\right\rfloor} \sum_{k | \gcd(i, j)}\mu(k) \\ =&\sum_{d = 1}^{M} [d \in P] \sum_{k = 1}^{\left\lfloor\frac{M}{d}\right\rfloor} \mu(k) \left\lfloor\frac{N}{dk}\right\rfloor \left\lfloor\frac{M}{dk}\right\rfloor \end{aligned}$

注意一个细节是 $\left\lfloor \frac{\left\lfloor \frac{N}{d} \right\rfloor}{k} \right\rfloor = \left\lfloor \dfrac{N}{dk} \right\rfloor$ 证明：
设 ${x\}$ 表示 $x$ 的小数部分： $\left\lfloor \frac{N}{dk} \right\rfloor = \left\lfloor \frac{ \left\lfloor \frac{N}{d} \right\rfloor + \left\{\frac{N}{d}\right\}}{k} \right\rfloor = \left\lfloor \frac{\left\lfloor \frac{N}{d} \right\rfloor}{k} + \frac{\left\{\frac{N}{d}\right\}}{k}\right\rfloor = \left\lfloor \frac{\left\lfloor \frac{N}{d} \right\rfloor}{k} \right\rfloor$

注意一个技巧：设 $T = d k$ ，于是可以枚举 $T$ ： $\sum_{T = 1}^M \left\lfloor\frac{N}{T}\right\rfloor \left\lfloor\frac{M}{T}\right\rfloor \sum_{d | T}[d \in P] \mu\left(\frac{T}{d}\right)$ 注意到 $\sum_{d | T}[d \in P] \mu\left(\frac{T}{d}\right)$ 可以 $\ln N)$ 预处理，查询的时候整数分块 $O(\sqrt N)$ 即可。

代码

要吸氧。

#include <bits/stdc++.h>

typedef long long LL;

const int MAXN = 10000000;

LL Num[MAXN + 5];

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];
		}
	}
	for (int i = 0; i < Primes.size(); i++) {
        int p = Primes[i];
        for (int j = p, c = 1; j <= n; j += p, c++)
            Num[j] += Mu[c];
	}
	for (int i = 1; i <= n; i++)
        Num[i] += Num[i - 1];
}

int main() {
    Init(MAXN);
    int T; scanf("%d", &T);
    while (T--) {
        int N, M; scanf("%d%d", &N, &M);
        if (N < M) std::swap(N, M); LL Ans = 0;
        for (LL lft = 1, rgt = 1; lft <= M; lft = rgt + 1) {
            rgt = std::min(M / (M / lft), N / (N / lft));
            Ans += (N / lft) * (M / lft) * (Num[rgt] - Num[lft - 1]);
        }
        printf("%lld\n", Ans);
    }
    return 0;
}