AcWing 221. 龙哥的问题（欧拉函数）

AcWing 221. 龙哥的问题

题意

给一个整数 $n(1\le n\le 2^{31})$ ，求 ${\sum }_{i=1}^{n}gcd(i,n)$

解法

枚举 $n$ 的因子。
$$\begin{array}{rl}ans& =\sum _{i=1}^{n}gcd(i,n)\\ & =\sum _{d}d\sum _{i=1}^n[gcd(i,n)==d]\\ & =\sum _{d}d\sum _{i=1}^n[gcd(i/d,n/d)==1]\\ & =\sum _{d}d\sum _{i=1}^{n/d}[gcd(i,n/d)==1]\\ & =\sum _{d}d\cdot \phi (n/d)\end{array}$$

线性枚举 $n$ 的因子，然后计算欧拉函数累加答案。
$1\cdot 10^9$ 范围的数最多因子个数大约是一千多，所以复杂度为 $O(1000\sqrt{n})$

代码

#pragma region
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <vector>
using namespace std;
typedef long long ll;
#define tr t[root]
#define lson t[root << 1]
#define rson t[root << 1 | 1]
#define rep(i, a, n) for (int i = a; i <= n; ++i)
#define per(i, a, n) for (int i = n; i >= a; --i)
#pragma endregion
const int maxn = 1e7 + 5;

int euler_phi(int n) {
    int m = int(sqrt(n + 0.5));
    int ans = n;
    for (int i = 2; i <= m; i++)
        if (n % i == 0) {
            ans = ans / i * (i - 1);
            while (n % i == 0) n /= i;
        }
    if (n > 1) ans = ans / n * (n - 1);
    return ans;
}
int main() {
    int n;
    scanf("%d", &n);
    ll ans = 0;
    ll i;
    for (i = 1; i * i < n; ++i) {
        if (n % i == 0) ans += 1LL * i * euler_phi(n / i) + 1LL * n / i * euler_phi(i);
    }
    if (1LL * i * i == n) ans += 1LL * i * euler_phi(i);
    printf("%lld\n", ans);
}