[CodeForces 547C] Mike and Foam（莫比乌斯反演） | 错题本

题目

[CodeForces 547C] Mike and Foam

分析

设 $c_i$ 表示 $i$ 的个数，要求 $i<j$ 比较麻烦，考虑求 $$t=\sum _{i=1}^n\sum _{j=1}^n[\gcd (a_i,a_j)=1]$$ 则答案为 $\frac{t-c_1}{2}$。设 $m$ 是最大值：$$\begin{array}{rl}t=& \sum _{i=1}^n\sum _{j=1}^n[\gcd (a_i,a_j)=1]\\ =& \sum _{d=1}^m\mu (d){\left(\sum _{i=1}^{\lfloor \frac{m}{d}\rfloor }{c}_{i\cdot d}\right)}^2\end{array}$$ 令 $f(x)=\sum _{i=1}^{\lfloor \frac{m}{x}\rfloor }{c}_{i\cdot x}$，发现修改一个数涉及到要修改的 $f$ 值是所有它的因数，枚举因数的时间复杂度是 $O(\sqrt{m})$，因此暴力修改即可。

最开始处理 $f$ 的复杂度是 $O({\sum }_{i=1}^m\frac{m}{i})=O(m\ln m)$，不会 T。

代码

#include <bits/stdc++.h>

typedef long long LL;

const int MAXN = 500000;

int N, M, Q;
int A[MAXN + 5];
LL Cnt[MAXN + 5], 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];
		}
	}
}

int main() {
	Init(MAXN);
	memset(Vis, 0, sizeof Vis);
	scanf("%d%d", &N, &Q);
	for (int i = 1; i <= N; i++)
		scanf("%d", &A[i]), M = std::max(M, A[i]);
	LL Ans = 0;
	while (Q--) {
		int pos; scanf("%d", &pos);
		int dlt = Vis[pos] ? -1 : 1;
		for (int i = 1; i * i <= A[pos]; i++) {
			if (A[pos] % i) continue;
			int j = A[pos] / i;
			if (j <= M / i) {
				Ans -= Num[i] * Num[i] * Mu[i];
				Num[i] += dlt;
				Ans += Num[i] * Num[i] * Mu[i];
			}
			if (i != j && i <= M / j) {
				Ans -= Num[j] * Num[j] * Mu[j];
				Num[j] += dlt;
				Ans += Num[j] * Num[j] * Mu[j];
			}
		}
		Cnt[A[pos]] += dlt;
		Vis[pos] = !Vis[pos];
		printf("%lld\n", (Ans - Cnt[1]) / 2);
	}
	return 0;
}