P1587 [NOI2016] 循环之美

P1587 [NOI2016] 循环之美

求 $\sum _{x=1}^n\sum _{y=1}^m\frac{x}{y}$ 在 $k$ 进制下能表示成循环节从第一位小数开始的无限循环小数或整数的最简分数个数。

先思考怎么转换。

首先肯定满足 $\gcd (x,y)=1$。

假设 $\frac{x}{y}$ 的循环节长度为 $l$，根据在 $k$ 进制下的数乘以 $k^p$ 相当于将小数点往后挪 $p$ 位，那么有：
{ x k l y } = { x y } \{\frac{xk^l}{y}\}=\{\frac{x}{y}\} {yxkl​}={yx​}
转换一下上面那个式子，有：
$$\begin{array}{rl}\frac{x{k}^l}{y}-\lfloor \frac{x{k}^l}{y}\rfloor & =\frac{x}{y}-\lfloor \frac{x}{y}\rfloor \\ x{k}^l-\lfloor \frac{x{k}^l}{y}\rfloor \cdot y& =x-\lfloor \frac{x}{y}\rfloor \cdot y\\ x{k}^l& \equiv x(\mathrm{m}\mathrm{o}\mathrm{d}y)\\ k^l& \equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}y)\\ \gcd (k,y)& =1\end{array}$$
那么题意就可以转换为求：
$$\sum _{i=1}^n\sum _{j=1}^m[\gcd (i,j)=1][\gcd (j,k)=1]$$
先化简原式，有：

$$\begin{array}{rl}& \sum _{i=1}^n\sum _{j=1}^m[\gcd (i,j)=1][\gcd (j,k)=1]\\ & =\sum _{i=1}^n\sum _{j=1}^m[\gcd (j,k)=1]\sum _{p|\gcd (i,j)}\mu (p)\\ & =\sum _{i=1}^n\sum _{j=1}^m[\gcd (j,k)=1]\sum _{p=1}^{\min (n,m)}\mu (p)[p|i][p|j]\\ & =\sum _{p=1}^{\min (n,m)}\mu (p)\sum _{i=1}^n\sum _{j=1}^m[p|i][p|j][\gcd (j,k)=1]\\ & =\sum _{p=1}^{\min (n,m)}\mu (p)\sum _{p|i}^n\sum _{p|j}^m[\gcd (j,k)=1]\\ & =\sum _{p=1}^{\min (n,m)}\mu (p)\lfloor \frac{n}{p}\rfloor \sum _{p|j}^m[\gcd (j,k)=1]\\ & =\sum _{p=1}^{\min (n,m)}\mu (p)[\gcd (p,k)=1]\lfloor \frac{n}{p}\rfloor \sum _{j=1}^{\lfloor \frac{m}{p}\rfloor }[\gcd (j,k)=1]\end{array}$$

再设个函数，并化简：
$$\begin{array}{r}f(n)=\sum _{i=1}^n[\gcd (i,k)=1]\end{array}$$

思考，当 $i>k$ 时，有 $\gcd (i,k)=\gcd (i+k,k)$，那么答案肯定是呈现一个长度为 $k$ 的循环，那么有：
$$\begin{array}{rl}f(n)& =f(n\mathrm{m}\mathrm{o}\mathrm{d}k)+\lfloor \frac{n}{k}\rfloor \phi (k)\end{array}$$
那么原式等于：
$$\sum _{p=1}^{\min (n,m)}\lfloor \frac{n}{p}\rfloor f(\lfloor \frac{m}{p}\rfloor )\mu (p)[\gcd (p,k)=1]$$
现在已经有整除分块了，然后是处理 $\sum _{i=1}^n\mu (i)[\gcd (i,k)=1]$ 前缀和的问题。

法一

设前面那个式子为 $s(n,k)$。

尝试化简 $s(n,k)$，则有：
$$\begin{array}{rl}s(n,k)& =\sum _{i=1}^n\mu (i)[\gcd (i,k)=1]\\ & =\sum _{i=1}^n\mu (i)\sum _{p=1}^{\min (n,k)}\mu (p)[p|i][p|k]\\ & =\sum _{p|k}^{\min (n,k)}\mu (p)\sum _{p|i}^n\mu (i)\\ & =\sum _{p|k}^{\min (n,k)}\mu (p)\sum _{i=1}^{\lfloor \frac{n}{p}\rfloor }\mu (ip)\\ & =\sum _{p|k}^{\min (n,k)}\mu (p{)}^2\sum _{i=1}^{\lfloor \frac{n}{p}\rfloor }\mu (i)[\gcd (i,p)=1]\\ & =\sum _{p|k}^{\min (n,k)}\mu (p{)}^{2}s(\lfloor \frac{n}{p}\rfloor ,p)\end{array}$$

然后数论分块即可。

法二

这里是设前面那个式子为 $s(n)$，则有：

$$\begin{array}{rl}s(n)& =\sum _{i=1}^n\mu (i)[\gcd (i,k)=1]\\ & =\sum _{i=1}^n[\gcd (i,k)=1]s(\lfloor \frac{n}{i}\rfloor )-\sum _{i=2}^n[\gcd (i,k)=1]s(\lfloor \frac{n}{i}\rfloor )\end{array}$$
先看前面那个：
$$\begin{array}{rl}& \sum _{i=1}^n[\gcd (i,k)=1]s(\lfloor \frac{n}{i}\rfloor )\\ & =\sum _{i=1}^n[\gcd (i,k)=1]\sum _{j=1}^{\lfloor \frac{n}{i}\rfloor }\mu (j)[\gcd (j,k)=1]\\ & =\sum _{i=1}^n\sum _{j=1}^{\lfloor \frac{n}{i}\rfloor }\mu (j)[\gcd (ij,k)=1]\\ & =\sum _{t=1}^n\sum _{d|t}[\gcd (t,k)=1]\mu (d)\\ & =\sum _{t=1}^n[\gcd (t,k)=1][t=1]\\ & =1\end{array}$$
那么最终有：
$$s(n)=1-\sum _{i=2}^n[\gcd (i,k)=1]s(\lfloor \frac{n}{i}\rfloor )$$
数论分块即可。

#include <bits/stdc++.h>

using namespace std;

const int _ = 1e6 + 5;

inline int read()
{
	int x = 0, f = 1;
	char c = getchar();
	while(c < '0' || c > '9')
	{
		if(c == '-') f = -1;
		c = getchar();
	}
	while(c >= '0' && c <= '9')
	{
		x = x * 10 + c - '0';
		c = getchar();
	}
	return x * f;
}

int n, m, k;

int g[2007], A[_];

int cnt, vis[_], pri[_], mu[_];

map<int, int> ans_A;

long long ans;

int f(int n)
{
	return g[n % k] + (n / k) * g[k];
}

void init()
{
	for(int i = 1; i <= k; ++i) g[i] = g[i - 1] + (__gcd(i, k) == 1);
	mu[1] = A[1] = 1;
	for(int i = 2; i <= _ - 5; ++i)
	{
		if(!vis[i])
		{
			mu[i] = -1;
			pri[++cnt] = i;
		}
		for(int j = 1; j <= cnt && i * pri[j] <= _ - 5; ++j)
		{
			int p = i * pri[j];
			vis[p] = 1;
			if(i % pri[j] == 0) break;
			mu[p] = -mu[i];
		}
		A[i] = A[i - 1] + mu[i] * (f(i) - f(i - 1));
	}
}

int F(int x)
{
	if(x <= _ - 5) return A[x];
	if(ans_A.find(x) != ans_A.end()) return ans_A[x];
	int ans = 1;
	for(int l = 2, r; l <= x; l = r + 1)
	{
		r = x / (x / l);
		ans -= F(x / l) * (f(r) - f(l - 1));
	}
	return ans_A[x] = ans;	
}

signed main()
{
	n = read(), m = read(), k = read();
	init();
	for(int l = 1, r; l <= min(n, m); l = r + 1)
	{
		r = min(n / (n / l), m / (m / l));
		ans += 1LL * (n / l) * f(m / l) * (F(r) - F(l - 1));
	}
	printf("%lld\n", ans);
	return 0;
}