【题解】BZOJ-4176 Lucas的数论

Lucas的数论

Description

• 给定整数 $n$，求
  $$\left[\sum _{i=1}^n\sum _{j=1}^{n}d(ij)\right]\mathrm{m}\mathrm{o}\mathrm{d}(10^9+7)$$

• 对于 $100\%$ 的数据 $n\le 10^9$。

Solution

$$\begin{array}{rlr}\sum _{i=1}^n\sum _{j=1}^{n}d(ij)& =\sum _{i=1}^n\sum _{j=1}^n\sum _{x\mid i}\sum _{y\mid j}[\gcd (x,y)=1]& \\ & =\sum _{x=1}^n\sum _{y=1}^n[\gcd (x,y)=1]\sum _{i=1}^n[x\mid i]\sum _{j=1}^n[y\mid j]& \\ & =\sum _{x=1}^n\sum _{y=1}^n[\gcd (x,y)=1]\lfloor \frac{n}{x}\rfloor \lfloor \frac{n}{y}\rfloor & \\ & =\sum _{d=1}^n\mu (d)\sum _{x=1}^n[d\mid x]\lfloor \frac{n}{x}\rfloor \sum _{y=1}^n[d\mid y]\lfloor \frac{n}{y}\rfloor & \\ & =\sum _{d=1}^n\mu (d){\left(\sum _{i=1}^{\lfloor \frac{n}{d}\rfloor }\lfloor \frac{n}{di}\rfloor \right)}^2& (1)\\ & =\sum _{d=1}^n\mu (d){\left(\sum _{i=1}^{\lfloor \frac{n}{d}\rfloor }d(i)\right)}^2& (2)\end{array}$$

$(1)$ 中有两层整除分块。

$\mu$ 用杜教筛，阈值为 $10^6$。

预处理 $d$ 的前缀和。

以 $k$ 为阈值，当 $\lfloor \frac{n}{d}\rfloor \le k$ 时用 $(2)$ 的公式，以预处理过的 $d$ 的前缀和来计算；当 $\lfloor \frac{n}{d}\rfloor >k$ 时用 $(1)$ 中的整除分块计算。

那么预处理是 $\mathrm{O}(k)$ 的。

第一层整除分块中，若 $\lfloor \frac{n}{d}\rfloor \le k$，即 $d\ge \lfloor \frac{n}{k}\rfloor$，那么就会用 $d$ 的前缀和 $\mathrm{O}(1)$ 计算，这一段是 $\mathrm{O}\left(\sqrt{n-\frac{n}{k}}\right)\le \mathrm{O}(\sqrt{n})$ 的。

第一层整除分块中，若 $\lfloor \frac{n}{d}\rfloor >k$，即 $d<\lfloor \frac{n}{k}\rfloor$，那么就会用整除分块来计算，一共是 ${\sum }_{d=1}^{\lfloor \frac{n}{k}\rfloor }\sqrt{\lfloor \frac{n}{d}\rfloor }$，即
$${\int }_0^{\frac{n}{k}}\sqrt{\frac{n}{x}}dx=\frac{2n}{\sqrt{k}}$$
也就是 $\mathrm{O}\left(\frac{n}{\sqrt{k}}\right)$ 的。

综上，总时间复杂度为 $\mathrm{O}\left(k+\sqrt{n}+\frac{n}{\sqrt{k}}\right)$，$k$ 大约取 $n^{\frac{2}{3}}$ 时最优，和杜教筛一样。

Code

// 18 = 9 + 9 = 18.
#include <iostream>
#include <cstdio>
#include <unordered_map>x
#define Debug(x) cout << #x << "=" << x << endl
typedef long long ll;
using namespace std;

const int MAXN = 1e6 + 5;
const int N = 1e6;
const int MOD = 1e9 + 7;

int p[MAXN], mu[MAXN], sum_mu[MAXN], d[MAXN], sum_d[MAXN], num[MAXN];
bool vis[MAXN];

void pre()
{
	mu[1] = sum_mu[1] = d[1] = sum_d[1] = 1;
	for (int i = 2; i <= N; i++)
	{
		if (!vis[i])
		{
			p[++p[0]] = i;
			mu[i] = -1;
			d[i] = 2;
			num[i] = 1;
		}
		for (int j = 1; j <= p[0] && i * p[j] <= N; j++)
		{
			vis[i * p[j]] = true;
			if (i % p[j] == 0)
			{
				mu[i * p[j]] = 0;
				d[i * p[j]] = d[i] / (num[i] + 1) * (num[i] + 2);
				num[i * p[j]] = num[i] + 1;
				break;
			}
			mu[i * p[j]] = mu[i] * mu[p[j]];
			d[i * p[j]] = d[i] * d[p[j]];
			num[i * p[j]] = 1;
		}
		sum_mu[i] = sum_mu[i - 1] + mu[i];
		sum_d[i] = (sum_d[i - 1] + d[i]) % MOD;
	}
}

unordered_map<int, int> dp_mu;

int sublinear_mu(int n)
{
	if (n <= N)
	{
		return sum_mu[n];
	}
	if (dp_mu.find(n) != dp_mu.end())
	{
		return dp_mu[n];
	}
	int res = 1;
	for (int l = 2, r; l <= n; l = r + 1)
	{
		int k = n / l;
		r = n / k;
		res = (res - (ll)(r - l + 1) * sublinear_mu(k) % MOD + MOD) % MOD;
	}
	return dp_mu[n] = res;
}

int getsum_mu(int l, int r)
{
	return (sublinear_mu(r) - sublinear_mu(l - 1) + MOD) % MOD;
}

int getsum_d(int n)
{
	if (n <= N)
	{
		return sum_d[n];
	}
	int res = 0;
	for (int l = 1, r; l <= n; l = r + 1)
	{
		int k = n / l;
		r = n / k;
		res = (res + (ll)(r - l + 1) * k % MOD) % MOD;
	}
	return res;
}

int block(int n)
{
	int res = 0;
	for (int l = 1, r; l <= n; l = r + 1)
	{
		int k = n / l;
		r = n / k;
		int tmp = getsum_d(k);
		res = (res + (ll)getsum_mu(l, r) * tmp % MOD * tmp % MOD) % MOD;
	}
	return res;
}

int main()
{
	pre();
	int n;
	scanf("%d", &n);
	printf("%d\n", block(n));
	return 0;
}