HDU5528 Count a*b 数论函数前缀和

Description

定义$f(m)={\sum }_{i=0}^{m-1}{\sum }_{j=0}^{m-1}[gcd(m,i\ast j)̸=m]$
求$g(n)={\sum }_{m|n}f(m)$

---

Input

The first line contains an integer $T$ indicating the total number of test cases. Each test case is a line with a positive integer $n$.
$1\le T\le 20000$
$1\le n\le 10^9$

---

Output

For each test case, print one integer $s$, representing $g(n)$ modulo $2^{64}$.

---

Solution

考虑$f(x)$的反面$h(x)$
$$f(m)=m^2-h(m)$$
$$h(m)=\sum _{i=0}^{m-1}\sum _{j=0}^{m-1}[gcd(m,i\ast j)=m]$$
那么对于$h(x)$进行变化。
考虑$i,j$的所有取值情况，在满足$gcd(m,ij)=m$的情况下二元组$(i,j)$有多少种。

---

我们考虑枚举$m$的所有因子$d$
如果$gcd(m,i)=d$，则要使得满足条件$gcd(m,ij)=m$，$j$需要满足条件$\frac{m}{d}|j$。
因为这样$j$中包含了$\frac{m}{d}$，相乘后$ij$中才能包含$m$，也就是$m|ij$。
以上也就是说满足$d|i,\frac{m}{d}|j$时有$m|ij$，则$gcd(m,ij)=m$。

---

那么对于$i$有 $0\le i<m$ 且$gcd(m,i)=d$
$i$取值可以为$k\ast d(1\le k\le \frac{m}{d})$，因为有$d|i$,所以$i=k\ast d$。
要满足
$$gcd(m,i)=d$$
$$gcd(m,k\ast d)=d$$
$$gcd(\frac{m}{d},k)=1$$
于是$k$的取值个数为$\phi (\frac{m}{d})$，这也是$i$的取值个数。

---

接下来考虑在满足上述条件下的$j$的所有可能取值。
$$\frac{m}{d}|j(0\le j<m)$$
则可以写成
$$j=k\ast \frac{m}{d}$$
于是
$$[k]=\frac{m}{\frac{m}{d}}=d$$
于是$k$的取值个数为$d$,这也是$j$的取值个数。

---

综上我们得到$h(m)$和$f(m)$的表达式
$$h(m)=\sum _{d|m}d\ast \phi (\frac{m}{d})$$
$$f(m)=m^2-h(m)=m^2-\sum _{d|m}d\ast \phi (\frac{m}{d})$$

---

那么接下来是$g(n)$
$$g(n)=\sum _{m|n}f(m)$$
$$g(n)=\sum _{m|n}{m}^2-\sum _{m|n}\sum _{d|m}d\ast \phi (\frac{m}{d})$$
令
$$g_1(n)=\sum _{m|n}{m}^2{g}_2(n)=\sum _{m|n}\sum _{d|m}d\ast \phi (\frac{m}{d})$$
可以发现的是$g_1(n)$和$g_2(n)$都是积性函数 (不是完全积性函数)。
那么满足$g(n)={\prod }_{i=0}^{c}g(p_i^{k_i})n=p_0^{k_0}.p_1^{k_1}...p_c^{k_c}$
$$g_1(n)=\prod _{i=0}^c{g}_1(p_i^{k_i})$$
$$g_2(n)=\prod _{i=0}^c{g}_2(p_i^{k_i})$$

---

其实对于求解 $g(p^k)$就变得简单了。
因为$p^k$只有$k+1$个因子$p^0,p^1,p^2,...,p^k$。
于是只有分解质因数的复杂度了。
$$g_1(p^k)=\sum _{m|n}{m}^2=\sum _{i=0}^k(p^{2i})$$
$$g_2(p^k)=\sum _{m|n}\sum _{d|m}d\ast \phi (\frac{m}{d})=\sum _{i=0}^k[p^i+\sum _{j=1}^i(p^j-p^{j-1})\ast p^{i-j}]$$
注:$\phi (p^k)=p^k-p^{k-1}$，上式子读者可手推便可以发现。
$$=\sum _{i=0}^k{p}^i+\sum _{i=0}^k\sum _{j=1}^i(p-1)\ast p^{j-1}\ast p^{i-j}$$
$$=\sum _{i=0}^k{p}^i+\sum _{i=0}^k\sum _{j=1}^i(p-1)\ast p^{i-1}$$
$$=\sum _{i=0}^k{p}^i+\sum _{i=0}^{k}i\ast (p-1)\ast p^{i-1}$$
$$=\sum _{i=0}^k{p}^i+\frac{p-1}{p}\sum _{i=0}^{k}i\ast p^i$$
$$=\sum _{i=0}^k{p}^i+\sum _{i=0}^{k}i\ast p^i-\sum _{i=0}^{k}i\ast p^{i-1}$$
$$=\sum _{i=0}^k(i+1)\ast p^i-\sum _{i=0}^{k}i\ast p^{i-1}$$
$$=\sum _{i=1}^{k+1}i\ast p^{i-1}-\sum _{i=0}^{k}i\ast p^{i-1}$$
$$=(k+1)\ast p^k-0=(k+1)\ast p^k$$

---

综上
$$g_1(n)=\prod _{i=0}^c{g}_1(p_i^{k_i})=\prod _{i=0}^c\sum _{j=0}^{k_i}(p_i^{2j})$$
$$g_2(n)=\prod _{i=0}^c{g}_2(p_i^{k_i})=\prod _{i=0}^c(k_i+1)p_i^{k_i}$$

---

Code

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+10;
typedef unsigned long long ll;
int prime[maxn];
inline void getPrime() {
    memset(prime,0,sizeof(prime));
    prime[1] = 1;
    for(int i=2;i<maxn;i++) {
        if(!prime[i]) prime[++prime[0]] = i; 
        for(int j=1;j<=prime[0] && prime[j]*i < maxn;j++) {
            prime[prime[j]*i] = 1;
            if(i % prime[j] == 0) break;
        }
    }
}
void getFactors(int x) {
    int tmp = x;
    ll res = 1,ans = 1;
    for(int i=1;i<=prime[0]&&prime[i]*prime[i]<=tmp;i++) {
        if(tmp % prime[i] == 0) {
            ll t1 = 1,t2 = 0,p = 1;
            while(tmp % prime[i] == 0) {
                tmp /= prime[i];
                p = p * prime[i];
                t1 = t1 + p * p;
                t2++;
            }
            //printf("%llu %llu\n",t1,t2);
            res = res * t1;
            ans = ans * (t2 + 1) * p;
        }
    }
    if(tmp != 1) {
        res = res * ((ll)tmp * tmp + 1);
        ans = ans * ((ll)2*tmp);
    }
    printf("%llu\n",res - ans);
}

int main()
{
    getPrime();
    int caset;scanf("%d",&caset);
    int n;
    while(caset--) {
        scanf("%d",&n);
        getFactors(n);
    }
    return 0;
}