POJ 2480 积性函数

题意

传送门 POJ 2480 Longge’s problem

题解

求 $$\sum _{1\le i\le n}gcd(i,n)$$ 暴力枚举显然超时，考虑到 $gcd(i,N)$ 的值域规模较小，将相同最大公约数的情况合并计算。$d=gcd(i,n)=d\times gcd(i/d,n/d)$，可得 $gcd(i/d,n/d)=1$，满足条件的个数为 $\varphi (n/d)$。则上式变换为 $$f(n)=\sum _{d|n}\varphi (\frac{n}{d})d=\sum _{d|n}\varphi (d)\frac{n}{d}$$ 直接枚举约数求解欧拉函数可以在时间限制内处理单次查询，但对于多次查询难以胜任。考虑积性函数，设 $n,m$ 互质，则有
$$f(nm)=\sum _{d|nm}\varphi (d)\frac{nm}{d}$$ 乘法分配律展开
$$f(nm)=\sum _{d=d_1{d}_2,d1|n,d2|m}\varphi (d_1{d}_2)\frac{nm}{d_1{d}_2}$$ 由于欧拉函数为积性函数，可得 $$f(nm)=(\sum _{d|n}\varphi (d)\frac{n}{d})\times (\sum _{d|m}\varphi (d)\frac{m}{d})=f(n)\times f(m)$$ 则 $f(n)$ 是积性函数。根据算数基本定理 $n=\prod _{i=1}^m{p}_i^{c_i}$，则有 $f(n)=\prod _{i=1}^{m}f(p_i^{c_i})$。对于单个质因子，有
$$f(p^c)=\sum _{d|p^c}\varphi (d)\frac{p^c}{d}=p^c+\sum _{1\le i\le c}{p}^{i-1}(p-1)\frac{p^c}{p^i}=(c+1)p^c-c{p}^{c-1}$$ 对 $n$ 进行质因子分解，则可以在 $O(\sqrt{n})$ 复杂度处理单次查询。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <map>
using namespace std;
typedef long long ll;
typedef map<int, int> mp;

mp prime_factor(int n)
{
    mp res;
    for (ll i = 2LL; i * i <= n; ++i)
    {
        while (n % i == 0)
            ++res[i], n /= i;
    }
    if (n > 1)
        ++res[n];
    return res;
}

int qpow(ll x, int n)
{
    int res = 1;
    while (n)
    {
        if (n & 1)
            res *= x;
        x *= x;
        n >>= 1;
    }
    return res;
}

int main()
{
    int n;
    while (~scanf("%d", &n))
    {
        mp prime = prime_factor(n);
        ll res = 1;
        for (mp::iterator it = prime.begin(); it != prime.end(); ++it)
        {
            int p = it->first, c = it->second, t = qpow(p, c - 1);
            res *= (ll)(c + 1) * p * t - (ll)c * t;
        }
        printf("%lld\n", res);
    }
    return 0;
}