【题目】HDU-6588 Function

我是万万没想到2022年了我居然还有机会写题解……

题目大意

求$\sum _{i=1}^n\gcd (\lfloor \sqrt[3]{i}\rfloor ,i)\mathrm{mod}998244353$。

共有$T$次询问，$1\le T\le 11$，$1\le n\le 10^{21}$。

---

思路

注意到$n$的上限$10^{21}$非常大（甚至爆了long long），枚举$i$显然是不可行的。但是式子中出现了$\lfloor \sqrt[3]{i}\rfloor$，如果枚举$\lfloor \sqrt[3]{i}\rfloor$，上限就变成了$10^7$，复杂度刚好可以接受。因此初步猜测每次询问的复杂度为$O(\sqrt[3]{n})$。

令$\lfloor \sqrt[3]{i}\rfloor =k$，得$k\le \sqrt[3]{i}<k+1$，即$k^3\le i<(k+1{)}^3$，因此原式化为

$$\sum _{i=1}^n\gcd (\lfloor \sqrt[3]{i}\rfloor ,i)=\sum _{k=1}^{\lfloor \sqrt[3]{n}\rfloor }\sum _{i=k^3}^{\min \{(k+1{)}^3-1,n\}}\gcd (k,i)$$

$k=\lfloor \sqrt[3]{n}\rfloor$时，$i$不能超过$n$，所以不能取遍$k^3$到$(k+1{)}^3-1$的值，需要单独处理。因此将上式中的$\min$展开：

$$\sum _{k=1}^{\lfloor \sqrt[3]{n}\rfloor -1}\sum _{i=k^3}^{(k+1{)}^3-1}\gcd (k,i)+\sum _{i=\lfloor \sqrt[3]{n}{\rfloor }^3}^n\gcd (\lfloor \sqrt[3]{n}\rfloor ,i)$$

这里需要使用一个结论：$\sum _{i=1}^n\gcd (a,i)=\sum _{T|a}\lfloor \frac{n}{T}\rfloor \phi (T)$。推导过程放在最后。

于是对上式进一步化简。前一部分化为

$$\begin{array}{rl}& \sum _{k=1}^{\lfloor \sqrt[3]{n}\rfloor -1}\left[\sum _{i=1}^{(k+1{)}^3-1}\gcd (k,i)-\sum _{i=1}^{k^3-1}\gcd (k,i)\right]\\ =& \sum _{k=1}^{\lfloor \sqrt[3]{n}\rfloor -1}\left[\sum _{T|k}\lfloor \frac{(k+1{)}^3-1}{T}\rfloor \phi (T)-\sum _{T|k}\lfloor \frac{k^3-1}{T}\rfloor \phi (T)\right]\\ =& \sum _{k=1}^{\lfloor \sqrt[3]{n}\rfloor -1}\sum _{T|k}\phi (T)\left[\lfloor \frac{(k+1{)}^3-1}{T}\rfloor -\lfloor \frac{k^3-1}{T}\rfloor \right]\\ =& \sum _{k=1}^{\lfloor \sqrt[3]{n}\rfloor -1}\sum _{T|k}\phi (T)\left[\left(\frac{k^3}{T}+3\frac{k^2}{T}+3\frac{k}{T}\right)-\left(\frac{k^3}{T}+\lfloor -\frac{1}{T}\rfloor \right)\right]\\ =& \sum _{k=1}^{\lfloor \sqrt[3]{n}\rfloor -1}\sum _{T|k}\phi (T)\left[3\frac{k^2}{T}+3\frac{k}{T}+1\right]\end{array}$$

枚举$T$的值，由于$k$是$T$的倍数，记$j=\frac{k}{T}$，则$1\le j\le \lfloor \frac{\lfloor \sqrt[3]{n}\rfloor -1}{T}\rfloor$，简记为$1\le j\le b$，上式进一步化为

$$\begin{array}{rl}& \sum _{T=1}^{\lfloor \sqrt[3]{n}\rfloor -1}\phi (T)\sum _{j=1}^b[3T{j}^2+3j+1]\\ =& \sum _{T=1}^{\lfloor \sqrt[3]{n}\rfloor -1}\phi (T)\left[3T\sum _{j=1}^b{j}^2+3\sum _{j=1}^{b}j+\sum _{j=1}^{b}1\right]\\ =& \sum _{T=1}^{\lfloor \sqrt[3]{n}\rfloor -1}\phi (T)\left[\frac{b(b+1)(2b+1)}{2}T+\frac{3b(b+1)}{2}+b\right]\end{array}$$

$T$的范围为$O(\sqrt[3]{n})$，用线性筛$O(\sqrt[3]{n})$预处理$\phi (T)$后，可在$O(\sqrt[3]{n})$的时间内枚举$T$算出上式。

还好我翻出了三年前的线性筛板子，不然这题没法做了。

后一部分化为

$$\begin{array}{rl}& \sum _{i=1}^n\gcd (\lfloor \sqrt[3]{n}\rfloor ,i)-\sum _{i=1}^{\lfloor \sqrt[3]{n}{\rfloor }^3-1}\gcd (\lfloor \sqrt[3]{n}\rfloor ,i)\\ =& \sum _{T|\lfloor \sqrt[3]{n}\rfloor }\lfloor \frac{n}{T}\rfloor \phi (T)-\sum _{T|\lfloor \sqrt[3]{n}\rfloor }\lfloor \frac{\lfloor \sqrt[3]{n}{\rfloor }^3-1}{T}\rfloor \phi (T)\\ =& \sum _{T|\lfloor \sqrt[3]{n}\rfloor }\phi (T)\left[\lfloor \frac{n}{T}\rfloor -\lfloor \frac{\lfloor \sqrt[3]{n}{\rfloor }^3-1}{T}\rfloor \right]\\ =& \sum _{T|\lfloor \sqrt[3]{n}\rfloor }\phi (T)\left[\lfloor \frac{n}{T}\rfloor -\left(\frac{\lfloor \sqrt[3]{n}{\rfloor }^3}{T}+\lfloor -\frac{1}{T}\rfloor \right)\right]\\ =& \sum _{T|\lfloor \sqrt[3]{n}\rfloor }\phi (T)\left[\lfloor \frac{n}{T}\rfloor -\frac{\lfloor \sqrt[3]{n}{\rfloor }^3}{T}+1\right]\end{array}$$

这里$T$的范围同样为$O(\sqrt[3]{n})$，不过由于$T$是$\lfloor \sqrt[3]{n}\rfloor$的因式，可在$O(\sqrt[6]{n})$的时间内枚举。

$n$的范围超过了long long，必要时可以使用__int128，但要谨慎使用，以防超时。

---

下面推导$\sum _{i=1}^n\gcd (a,i)=\sum _{T|a}\lfloor \frac{n}{T}\rfloor \phi (T)$。枚举$\gcd (a,i)$的值$d$：

$$\sum _{i=1}^n\gcd (a,i)=\sum _{d|a}d\sum _{i=1}^n[\gcd (a,i)=d]$$

其中$\gcd (a,i)=d$等价于$\gcd \left(\frac{a}{d},\frac{i}{d}\right)=1$，记$k=\frac{i}{d}$，则$1\le k\le \lfloor \frac{n}{d}\rfloor$，上式化为

$$\sum _{d|a}d\sum _{k=1}^{\lfloor \frac{n}{d}\rfloor }\left[\gcd \left(\frac{a}{d},k\right)=1\right]$$

利用$\epsilon (n)=[n=1]=\sum _{d|n}\mu (d)$，上式转化为

$$\begin{array}{rl}& \sum _{d|a}d\sum _{k=1}^{\lfloor \frac{n}{d}\rfloor }\epsilon \left(\gcd \left(\frac{a}{d},k\right)\right)\\ =& \sum _{d|a}d\sum _{k=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{td|a,t|k}\mu (t)\end{array}$$

交换$k,t$的枚举顺序，由于$k$是$t$的倍数，记$j=\frac{k}{t}$，则$1\le j\le \lfloor \lfloor \frac{n}{d}\rfloor /t\rfloor =\lfloor \frac{n}{td}\rfloor$，进而

$$\begin{array}{rl}& \sum _{d|a}d\sum _{td|a}\sum _{j=1}^{\lfloor \frac{n}{td}\rfloor }\mu (t)\\ =& \sum _{d|a}d\sum _{td|a}\mu (t)\lfloor \frac{n}{td}\rfloor \end{array}$$

再利用$\phi (n)=\sum _{d|n}d\mu \left(\frac{n}{d}\right)$，令$td=T$，则$t=\frac{T}{d}$，所以

$$\begin{array}{rl}& \sum _{d|a}d\sum _{T|a,d|T}\mu \left(\frac{T}{d}\right)\lfloor \frac{n}{T}\rfloor \\ =& \sum _{T|a}\lfloor \frac{n}{T}\rfloor \sum _{d|T}d\mu \left(\frac{T}{d}\right)\\ =& \sum _{T|a}\lfloor \frac{n}{T}\rfloor \phi (T)\end{array}$$