[RC - 02] GCD（莫比乌斯反演 + 杜教筛） | 错题本

最新推荐文章于 2022-11-11 21:20:33 发布

ixRic

最新推荐文章于 2022-11-11 21:20:33 发布

阅读量208

点赞数 1

分类专栏： [ 错题本 ] # 杜教筛 # 莫比乌斯反演文章标签：数论莫比乌斯函数杜教筛

本文链接：https://blog.csdn.net/C20190102/article/details/108478893

版权

[ 错题本 ] 同时被 3 个专栏收录

86 篇文章 1 订阅

订阅专栏

莫比乌斯反演

13 篇文章 0 订阅

订阅专栏

杜教筛

3 篇文章 0 订阅

订阅专栏

文章目录

题目
分析
代码

题目

[RC - 02] GCD

分析

枚举 $p$ 到 $\left\lfloor\frac{N}{j}\right\rfloor$ 实际上就是在枚举 $j p$ ，于是原式变成
$\begin{aligned}&\sum_{i = 1}^N\sum_{j = 1}^N\sum_{p = 1}^{N}\sum_{q = 1}^{N}[\gcd(i, j) = 1][\gcd(p, q) = j] \\= &\sum_{i = 1}^N\sum_{j = 1}^N\sum_{p = 1}^{N}\sum_{q = 1}^{N}[\gcd(i, p, q) = 1] \\=&\sum_{d = 1}^{N}\left\lfloor\frac{N}{d}\right\rfloor^3\mu(d)\end{aligned}$ 整数分块 + 杜教筛即可。

代码

#include <bits/stdc++.h>

typedef long long LL;

const int MAXN = 1000000;
const int MOD = 998244353;

int Mu[MAXN + 5];
bool Vis[MAXN + 5];
std::vector<int> Primes;
std::unordered_map<int, int> Sum;

void Init(int n) {
    Mu[1] = 1;
    for (int i = 2; i <= n; i++) {
        if (!Vis[i])
            Mu[i] = -1, Primes.push_back(i);
        for (int j = 0; j < (int)Primes.size() && i * Primes[j] <= n; j++) {
            Vis[i * Primes[j]] = true;
            if (i % Primes[j] == 0) {
                Mu[i * Primes[j]] = 0;
                break;
            }
            Mu[i * Primes[j]] = -Mu[i];
        }
        Mu[i] = ((Mu[i - 1] + Mu[i]) % MOD + MOD) % MOD;
    }
}

int GetSum(int n) {
    if (n <= MAXN) return Mu[n];
    if (Sum.count(n)) return Sum[n];
    int res = 1;
    for (int lft = 2, rgt = 1; lft <= n; lft = rgt + 1) {
        rgt = n / (n / lft);
        res = (res - (LL)(rgt - lft + 1) * GetSum(n / lft) % MOD) % MOD;
    }
    return Sum[n] = (res + MOD) % MOD;
}

int main() {
    Init(MAXN);
    int N, Ans = 0; scanf("%d", &N);
    for (int lft = 1, rgt = 1; lft <= N; lft = rgt + 1) {
        rgt = N / (N / lft);
        Ans = (Ans + ((LL)(GetSum(rgt) - GetSum(lft - 1)) * (N / lft) % MOD * (N / lft) % MOD * (N / lft) % MOD)) % MOD;
    }
    printf("%d", (Ans + MOD) % MOD);
    return 0;
}