ZOJ 3881 From the ABC conjecture

题目：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3881

题意：
问题和ABC猜想无关，定义$rad(n)$表示$n$的因子中最大的无平方因子数，$\varphi(n)$表示不大于$n$且与$n$互质的正整数个数，令$f(n)=rad(n)\cdot\varphi(\frac{n}{rad(n)}),g(n)=\sum_{d|n}{f(d)},h(n)=\sum_{i=1}^{n}{g(i)}$，给定正整数$N$，求$h(N)\mod (10^9+9)$的值。
$n\le 10^{12}$。

题解：
若$n=\prod_{i=1}^{t}{p_i^{k_i}}$，则$rad(n)=\prod_{i=1}^{t}{p_i}$，是一个积性函数。
而对于$(n,m)=1$，显然有$(\frac{n}{rad(n)},\frac{m}{rad(m)})=1$，则$f(n\cdot m)=rad(n\cdot m)\cdot\varphi(\frac{n\cdot m}{rad(n\cdot m)})=rad(n)\cdot rad(m)\cdot\varphi(\frac{n}{rad(n)})\cdot\varphi(\frac{m}{rad(m)})=f(n)\cdot f(m)$，也是一个积性函数。
那么$g(n)=\sum_{d|n}{f(d)}=\prod_{i=1}^{t}{\sum_{j=0}^{k_i}{f(p_i^j)}}=\prod_{i=1}^{t}{(1+p_i\cdot\sum_{j=1}^{k_i}{\varphi(p_i^{j-1})})}=\prod_{i=1}^{t}{(p_i^{k_i}+1)}$。
仔细看$g(n)$的式子，相当于选出了一些$n$的约数$d$进行求和，而且每个质因数$p_i$要么不选，要么选$p_i^{k_i}$，这样的$d$是满足$(d,\frac{n}{d})=1$的，因此$g(n)=\sum_{d|n}{[(d,\frac{n}{d})=1]\cdot d}=\sum_{i=1}^{n}\sum_{j=1}^{n}{[(i,j)=1]\cdot[ij=n]\cdot i}$。
那么有$h(n)=\sum_{k=1}^{n}\sum_{i=1}^{n}\sum_{j=1}^{n}{[(i,j)=1]\cdot[ij=k]\cdot i}=\sum_{i=1}^{n}\sum_{j=1}^{n}{[(i,j)=1]\cdot[ij\le n]\cdot i}$。
所以有

$$h(n)=\sum_{i=1}^{n}\sum_{j=1}^{n}{[(i,j)=1][ij\le n]i}$$
利用 $[n=1]=\sum_{d|n}{\mu(d)}$得 $$h(n)=\sum_{i=1}^{n}\sum_{j=1}^{n}{[ij\le n]i\sum_{d}{[d|i][d|j]\mu(d)}}$$
而当 $ij\le n$时， $d\le\min(i,j)\le\sqrt{n}$，故有 $$h(n)=\sum_{d=1}^{\sqrt{n}}{d\mu(d)\sum_{i=1}^{n}\sum_{j=1}^{n}{[d|i][d|j][ij\le n]\frac{i}{d}}}\\=\sum_{d=1}^{\sqrt{n}}{d\mu(d)\sum_{i=1}^{\lfloor\frac{n}{d}\rfloor}\sum_{j=1}^{\lfloor\frac{n}{d}\rfloor}{[ijd^2\le n]i}}$$
可以发现 $j$很容易被消除，$i$有更严格的上界，所以 $$h(n)=\sum_{d=1}^{\sqrt{n}}{d\mu(d)\sum_{i=1}^{\left\lfloor \frac{n}{d^2} \right\rfloor}{\left\lfloor\frac{n}{i\cdot d^2} \right\rfloor i}} =\sum_{d=1}^{\sqrt{n}}{d\mu(d)s(\left\lfloor\frac{n}{d^2} \right\rfloor)}$$
其中 $s(n)=\sum_{i=1}^{n}{i\cdot\left\lfloor\frac{n}{i}\right\rfloor}=\sum_{i=1}^{n}{\sigma(i)}$可以 $O(\sqrt{n})$计算。
注意到这里复杂度下界已经是 $O(\sqrt{n})$，所以枚举 $d$即可，不需要根据$\left\lfloor\frac{n}{d^2}\right\rfloor$进行分段计算。
设计算 $h(n)$的时间复杂度为 $T(n)$，则 $$T(n)=O(\sqrt{n}+\sum_{d=1}^{\sqrt{n}}{\sqrt{\frac{n}{i^2}}})=O(\sqrt{n}\sum_{d=1}^{\sqrt{n}}{\frac{1}{i}})=O(\sqrt{n}\log n)$$

代码：

#include <cmath>
#include <cstdio>
typedef long long LL;
const int maxn = 1000001, mod = 1000000009, inv2 = 500000005;
int f[maxn];
int s(LL n)
{
    int res = 0;
    for(LL i = 1, j; i <= n; i = j + 1)
    {
        j = n / (n / i);
        res = (res + n / i % mod * ((j - i + 1) % mod) % mod * ((i + j) % mod)) % mod;
    }
    res = (LL)res * inv2 % mod;
    return res;
}
int main()
{
    f[1] = 1;
    for(int i = 1; i < maxn; ++i)
    {
        for(int j = i + i; j < maxn; j += i)
            f[j] -= f[i];
        f[i] *= i;
    }
    for(LL n; scanf("%lld", &n) == 1; )
    {
        int ans = 0, lim = (int)ceil(sqrt(n));
        for(int i = 1; i <= lim; ++i)
            if(f[i])
                ans = (ans + (LL)f[i] * s(n / i / i)) % mod;
        if(ans < 0)
            ans += mod;
        printf("%d\n", ans);
    }
    return 0;
}