牛客练习赛76 phi and phi 莫比乌斯反演+差分

最新推荐文章于 2022-10-23 20:24:00 发布

kaka0010

最新推荐文章于 2022-10-23 20:24:00 发布

阅读量223

点赞数 1

分类专栏： # 数论

本文链接：https://blog.csdn.net/kaka03200/article/details/114453448

版权

数论专栏收录该内容

22 篇文章 0 订阅

订阅专栏

原题链接：https://ac.nowcoder.com/acm/contest/10845/F

题意

$ans(n)=\sum_{i=1}^{n}\sum_{j=1}^{n}\phi(ij)\phi(gcd(i,j))$

化简

先引入一个定理
$\phi(ij)=\frac{\phi(i)\phi(j)gcd(i,j)}{\phi(gcd(i, j))}$

然后就变成非常一般的形式

$ans(n)=\sum_{i=1}^{n}\sum_{j=1}^{n}\phi(i)\phi(j)gcd(i, j)$

$=\sum_{d=1}^{n}d\sum_{i=1}^{n}\sum_{j=1}^{n}\phi(i)\phi(j)[gcd(i, j)=d]$

$=\sum_{d=1}^{n}d\sum_{i=1}^{n/d}\sum_{j=1}^{n/d}\phi(id)\phi(jd)[gcd(i, j)=1]$

$=\sum_{d=1}^{n}d\sum_{i=1}^{n/d}\sum_{j=1}^{n/d}\phi(id)\phi(jd)\sum_{t|i,t|j}\mu(t)$

$=\sum_{d=1}^{n}d\sum_{t=1}^{n/d}\mu(t)\sum_{i=1}^{n/td}\sum_{j=1}^{n/td}\phi(idt)\phi(jdt)$

$设 T = t d$

$=\sum_{T=1}^{n}\sum_{i=1}^{n/T}\phi(iT)\sum_{j=1}^{n/T}\phi(jT)\sum_{d|T}d\mu(\frac{T}{d})$

$\sum_{i=1}^{n/T}\phi(iT)$

$=\sum_{T=1}^{n}F(n,d)^2\phi(T)$

接着就可以直接写了，同样使用差分，考虑F函数对不同n的贡献，把两端点算出来差分一下就可以了

Code

#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define re register
typedef long long ll;
typedef pair<int, int> PII;
const int N = 1e6 + 100, M = 5e5 + 5, INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
ll prime[N], mu[N], k, ans[N];
ll phi[N];
bool is_prime[N];
inline void init(int n) {
    memset(is_prime, true, sizeof is_prime);
    mu[1] = 1; phi[1] = 1;
    for (int i = 2; i < n; ++i) {
        if (is_prime[i]) prime[++k] = i, mu[i] = -1, phi[i] = i-1;
        for (int j = 1; j <= k && i * prime[j] < n; ++j) {
            is_prime[i * prime[j]] = false;
            if (i % prime[j] == 0) {
                phi[i * prime[j]] = phi[i] * prime[j];
                break;
            } else {
                mu[i * prime[j]] = -mu[i];
                phi[i * prime[j]] = phi[i] * (prime[j] - 1);
            }
        }
    }
    for (int T = 1; T < n; T++) {
        ll tmp = 0;
        for (int i = T; i < n; i += T) {
            int l = i;
            int r = min(n, l + T - 1);
            tmp = (tmp + phi[i]) % MOD;
            ans[l] = (ans[l] + tmp * tmp % MOD * phi[T] % MOD) % MOD;
            ans[r + 1] = (ans[r + 1] - tmp * tmp % MOD * phi[T] % MOD + MOD) % MOD;
        }
    }
    for (int i = 1; i < n; i++) ans[i] = ((ans[i] + ans[i-1]) % MOD + MOD) % MOD;
}

void solve() {
    init(N);
    int n; cin >> n;
    for (int i = 1; i <= n; i++) printf("%lld\n", ans[i]);
}

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
#ifdef ACM_LOCAL
    freopen("input", "r", stdin);
    freopen("output", "w", stdout);
#endif
    solve();
}