【UOJ #221】【NOI 2016】循环之美

http://uoj.ac/problem/221
因为\(a\)和\(b\)不互质时，\(\frac ab=\frac{\frac a{(a,b)}}{\frac b{(a,b)}}\)，所以只用求\(a\)和\(b\)互质时的满足条件的个数。
\(\frac ab\)在\(k\)进制下是纯循环小数，我们先假设循环节长度为\(l\)，这样\(\frac ab\times k^l-\frac ab\)的小数部分就是0，也就是这是个整数。
\(\frac{a\left(k^l-1\right)}b\)是个整数，就是说\(b|a\left(k^l-1\right)\)。又因为\(a\bot b\)，所以\(b|\left(k^l-1\right)\)。
判断\(\frac ab\)是否是\(k\)进制下的纯循环小数就转变成了判断是否存在一个\(l,l\geq 0\)，满足\(\left(k^l-1\right)\mod b=0\)。
当\(k\)和\(b\)不互质时，对于所有的\(l\)，\(b\)和\(k^l\)有共同的质因子，又因为\(k^l\bot\left(k^l-1\right)\)，所以\(b\)和\(k^l\)的共同的质因子是\(k^l-1\)没有的，所以不存在\(l\)满足\(b|\left(k^l-1\right)\)。
当\(k\)和\(b\)互质时，由欧拉定理：\(k^{\varphi(b)}\mod b=1\)，存在\(l=\varphi(b)\)。
这样答案就变成了：$$\sum_{a=1}^n\sum_{b=1}m[b\bot k][a\bot b]$$
化一波式子：

\[\begin{aligned} &\sum_{a=1}^n\sum_{b=1}^m[b\bot k][a\bot b]\\ =&\sum_{d=1}^n\mu(d)\sum_{a=1}^{\left\lfloor\frac nd\right\rfloor}\sum_{b=1}^{\left\lfloor\frac md\right\rfloor}[bd\bot k]\\ =&\sum_{d=1}^n\mu(d)[d\bot k]\left\lfloor\frac nd\right\rfloor\sum_{b=1}^{\left\lfloor\frac md\right\rfloor}[b\bot k] \end{aligned} \]

\(\sum\limits_{i=1}^n[i\bot k]\)很好预处理后\(O(1)\)计算。
重点是怎么算\(\sum\limits_{d=1}^n\mu(d)[d\bot k]\)？
先放宽限制，把\([d\bot k]\)的限制去掉，这样就是对\(O\left(\sqrt n+\sqrt m\right)\)个下取整取值求\(\mu\)的前缀和，可以先\(O\left(n^{\frac 23}\right)\)大力杜教筛一波。
然后考虑\(S(i,n)\)表示\(1\sim n\)中与\(k\)的前\(i\)个质因子互质的数的\(\mu\)值和，这样\(S(i,n)=S(i-1,n)-\mu\left(p_i\right)S\left(i,\left\lfloor\frac n{p_i}\right\rfloor\right)\)递推求出，时间复杂度\(O\left(\left(\sqrt n+\sqrt m\right)\log k\right)\)（假设\(k\)的不同质因子有\(\log k\)个）。
最后对\(O\left(\sqrt n+\sqrt m\right)\)个不同的\(\left\lfloor\frac nd\right\rfloor\)和\(\left\lfloor\frac md\right\rfloor\)的取值进行分段求和即可。
时间复杂度\(O\left(n^{\frac 23}+\left(\sqrt n+\sqrt m\right)\log k\right)\)。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;

const int N = 1000003;

bool notp[N];
int mu[N], sum_mu[N], prime[N], Num = 0;

void Euler_shai() {
	mu[1] = sum_mu[1] = 1;
	for (int i = 2; i <= 1000000; ++i) {
		if (!notp[i]) prime[++Num] = i, mu[i] = -1;
		for (int j = 1; j <= Num && prime[j] * i <= 1000000; ++j) {
			notp[prime[j] * i] = true;
			if (i % prime[j] == 0) break;
			mu[prime[j] * i] = -mu[i];
		}
		sum_mu[i] = sum_mu[i - 1] + mu[i];
	}
}

int R[N], cnt = 0, sum_mu2[N], n, m, k, S[N], pr[N], prnum = 0;

const int mo = 2333333;

struct HashTable {
	int pos[mo], num[mo];
	
	void ins(int nu, int po) {
		int tmp = nu % mo;
		while (num[tmp]) {++tmp; if (tmp == mo) tmp = 0;}
		num[tmp] = nu;
		pos[tmp] = po;
	}
	
	int query(int nu) {
		int tmp = nu % mo;
		while (num[tmp] != nu) {++tmp; if (tmp == mo) tmp = 0;}
		return pos[tmp];
	}
} HT;

int sum[N], id[N];

int gcd(int a, int b) {return b ? gcd(b, a % b) : a;}

void pre_sum() {
	for (int i = 1; i <= k; ++i) {
		sum[i] = sum[i - 1];
		if (gcd(i, k) == 1)
			++sum[i];
	}
}

int cal(int num) {return sum[k] * (num / k) + sum[num % k];}

int Sum(int num) {return num <= 1000000 ? S[id[num]] : S[HT.query(num)];}

int main() {
	scanf("%d%d%d", &n, &m, &k);
	Euler_shai();
	
	for (int i = 1, j = 1; i <= n && i <= m; ++i) {
		j = m / (m / i);
		i = n / (n / i);
		if (i > j) i = j;
		R[++cnt] = i;
		if (i > 1000000) HT.ins(i, cnt);
	}
	
	for (int i = 1; i <= cnt; ++i) {
		int num = R[i], &ret = S[i];
		if (num <= 1000000) {ret = sum_mu[num]; id[num] = i; continue;}
		ret = 1;
		for (int j = 2, pre = 1; j <= num; pre = j, ++j) {
			j = num / (num / j);
			ret -= 1ll * Sum(num / j) * (j - pre);
		}
	}
	
	for (int i = 1; i <= Num && prime[i] <= k; ++i)
		if (k % prime[i] == 0)
			pr[++prnum] = prime[i];
	
	for (int i = 1, pi = pr[1]; i <= prnum; pi = pr[++i])
		for (int j = 1; j <= cnt; ++j)
			S[j] += Sum(R[j] / pi);
	
	pre_sum();
	
	ll ans = 0;
	for (int tmp = 1, i = 1, j = 1; i <= n && i <= m; ++tmp, ++i) {
		j = m / (m / i);
		i = n / (n / i);
		if (i > j) i = j;
		ans += 1ll * (S[tmp] - S[tmp - 1]) * (n / i) * cal(m / i);
	}
	
	printf("%lld\n", ans);
	return 0;
}