HDU 5528 Count a * b （2015长春站B题&&积性函数）

设$h[n]$为$a*b=0$的个数。

$$f[n]=n^2-h[n]\\ g[n]=\sum_{m|n}f[m]=\sum_{m|n}{(m^2-h[m])}=\sum_{m|n}{m^2}-\sum_{m|n}{h[m]}$$

对于$h[n]$

$$h[n]=\sum_{i=1}^{n}\sum_{j=1}^{n}(gcd(i*j,n)=n)=\sum_{i=1}^{n}\sum_{j|{n\over{gcd(i,n)}},1<=j<=n}1=\sum_{i=1}^{n}gcd(i,n)\\ =\sum_{i=1}^{n}\sum_{d|gcd(i,n)}\phi(d)=\sum_{d|n}\phi(d)\sum_{d|i,1<=i<=n}1=\sum_{d|n}\phi(d)*{n\over{d}}$$

显然$h[n]$是积性函数，$\sum_{m|n}{h[m]}$也是积性函数。

对于$n=p^{\alpha}$时。

$$\sum_{m|n}{h[m]}=\sum_{m|n}\sum_{d|m}\phi(d)*{n\over{d}}=\sum_{i=0}^{\alpha}(p^{i}+\sum_{j=1}^{i}(p^j-p^{j-1})p^{i-j})\\ =\sum_{i=0}^{\alpha}p^i+{{p-1\over{p}}}\sum_{i=0}^{\alpha}i*p^i=(\alpha+1)p^{\alpha}$$

最后一步化简：错位相消。

设$n=p_1^{\alpha_1}p_2^{\alpha_2}...p_k^{\alpha_k}$

$$\sum_{m|n}{h[m]}=(\alpha_1+1)*(\alpha_2+1)*...*(\alpha_k+1)*p_1^{\alpha_1}*p_2^{\alpha_2}*...*p_k^{\alpha_k}=\phi(n)*n$$

所以：

$$g[n]=\sum_{m|n}m^2-n*\phi(n)$$

暴力枚举m的值，总的复杂度是$o(\sqrt{n})$。

小$trick$：
题目中要求对$2^{64}$取模，实际上就是用$unsigned \ long \ long$运算。

代码：

#include <bits/stdc++.h>
#define uLL unsigned long long
#define LL long long
#define FOR(i,x,y)  for(int i = x;i < y;++ i)
#define IFOR(i,x,y) for(int i = x;i > y;-- i)

using namespace std;

const int maxn = 100010;
int prime[maxn];
bool check[maxn];

void Get_Prime(){
    memset(check,false,sizeof(check));
    prime[0] = 0;
    FOR(i,2,maxn){
        if(!check[i]){
            prime[++prime[0]] = i;
        }
        FOR(j,1,prime[0]+1){
            if(i*prime[j] > maxn)   break;
            check[i*prime[j]] = true;
            if(i%prime[j] == 0) break;
        }
    }
}

vector <uLL> fac;
int p[30],p_len,cnt[30];

void dfs(int wei,int u){
    if(wei >= p_len)    {fac.push_back(u);  return;}
    dfs(wei+1,u);
    int q = p[wei];
    FOR(i,0,cnt[wei]){
        dfs(wei+1,u*q);
        q = q*p[wei];
    }
}

int n;

void work(){
    uLL res = (uLL)n;
    p_len = 0;
    memset(cnt,0,sizeof(cnt));
    FOR(i,1,prime[0]+1){
        if(prime[i] > n/prime[i])   break;
        if(n%prime[i] == 0) p[p_len++] = prime[i];
        while(n%prime[i] == 0)  {cnt[p_len-1] ++;n /= prime[i];}
    }
    if(n > 1)   {p[p_len++] = n;cnt[p_len-1] = 1;}
    FOR(i,0,p_len){
        res *= ((uLL)cnt[i]+1);
    }
    fac.clear();
    dfs(0,1);
    uLL ans = 0;
    FOR(i,0,(int)fac.size()){
        uLL v = fac[i];
        ans += v*v;
    }
    ans = ans - res;
    cout<<ans<<endl;
}

int main(){
    //freopen("test.in","r",stdin);
    Get_Prime();
    int T;  scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        work();
    }
    return 0;
}