[CodeForces 839D] Winter is here（莫比乌斯反演） | 错题本

最新推荐文章于 2021-07-14 10:32:28 发布

ixRic

最新推荐文章于 2021-07-14 10:32:28 发布

阅读量192

点赞数 1

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

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

版权

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

86 篇文章 1 订阅

订阅专栏

莫比乌斯反演

13 篇文章 0 订阅

订阅专栏

文章目录

题目
分析
代码

题目

[CodeForces 839D] Winter is here

分析

设 $m$ 是最大的数，枚举 $\gcd = d$ ，子序列长度为 $l$ ，要求的就是 $\begin{aligned} &\sum_{d = 2}^{m}\sum_{l = 1}^{n} d\cdot l \sum_{1 \leq k_1 < k_2 < \cdots < k_l \leq n} \left[\gcd_{i = 1}^l a_{k_i} = d\right] \\ = &\sum_{d = 2}^{m}\sum_{l = 1}^{n} d\cdot l \sum_{1 \leq k_1 < k_2 < \cdots < k_l \leq n} \left(\prod_{i = 1}^{l} [d | a_{k_i}] \right)\left[\gcd_{i = 1}^{l} \frac{a_{k_i}}{d} = 1\right]\end{aligned}$ 用莫比乌斯函数的基本性质 $\mu * 1 = \varepsilon$ ，枚举 $\frac{a_{k_i}}{d}$ 的因数上式转化为： $\begin{aligned} &\sum_{d = 2}^{m}\sum_{l = 1}^{n} d\cdot l\sum_{p = 1}^{\left\lfloor \frac{m}{d} \right\rfloor} \mu(p)\sum_{1 \leq k_1 < k_2 < \cdots < k_l \leq n} \prod_{i = 1}^{l}[(d\cdot p) | a_{k_i}] \\ =&\sum_{d = 2}^{m} d \sum_{p = 1}^{\left\lfloor \frac{m}{d} \right\rfloor} \mu(p) \sum_{l = 1}^{n} l \sum_{1 \leq k_1 < k_2 < \cdots < k_l \leq n} \prod_{i = 1}^{l}[(d\cdot p) | a_{k_i}]\end{aligned}$ 记录原数组中 $d\cdot p$ 倍数的数量 $c_{d\cdot p}$ ，然后 $\begin{aligned} \sum_{l = 1}^{n} l \sum_{1 \leq k_1 < k_2 < \cdots < k_l \leq n} \prod_{i = 1}^{l}[(d\cdot p) | a_{k_i}] = \sum_{l = 1}^{c_{d \cdot p}} l\binom{c_{d\cdot p}}{l}\end{aligned}$

令 $\sum_{i = 1}^{n} i\binom{n}{i}$ ，由于 $\binom{n}{i} = \binom{n - 1}{i - 1} + \binom{n - 1}{i}$ ，所以 $\begin{aligned}f(n) = &\sum_{i = 1}^{n} i\binom{n}{i} \\ =&\sum_{i = 1}^n i \binom{n - 1}{i - 1} + \sum_{i = 1}^n i\binom{n - 1}{i} \\ =&\sum_{i = 0}^{n - 1}\binom{n - 1}{i} + 2f(n - 1) \\ = &2f(n - 1) + 2^{n - 1}\end{aligned}$ 能够线性递推。
预处理 $f, c$ 等套式子计算即可，时间复杂度 $\ln m)$ 。

代码

#include <bits/stdc++.h>

typedef long long LL;

const int MAXN = 200000;
const int MAXM = 1000000;
const int MOD = 1000000007;

inline int Add(const int &x, const int &y) {
    return (x + y >= MOD) ? (x + y - MOD) : (x + y);
}

inline int Mul(const int &x, const int &y) {
    return (LL)x * y % MOD;
}

inline int Sub(const int &x, const int &y) {
    return (x - y < 0) ? (x - y + MOD) : (x - y);
}

int Pow(int x, int y) {
    int ret = 1;
    while (y) {
        if (y & 1)
            ret = Mul(ret, x);
        y >>= 1;
        x = Mul(x, x);
    }
    return ret;
}

int N, M;
int A[MAXN + 5];
int Cnt[MAXM + 5], Num[MAXN + 5];

int Mu[MAXM + 5];
bool Vis[MAXM + 5];
std::vector<int> Primes;

void Init(int n) {
	Mu[1] = 1;
	for (int i = 2; i <= n; i++) {
		if (!Vis[i])
			Mu[i] = Sub(0, 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]] = Sub(0, Mu[i]);
		}
	}
}

int main() {
    Init(MAXM);
    scanf("%d", &N);
    for (int i = 1; i <= N; i++) {
        scanf("%d", &A[i]);
        Cnt[A[i]]++, M = std::max(M, A[i]);
    }
    for (int i = 1; i <= M; i++)
        for (int j = i + i; j <= M; j += i)
            Cnt[i] = Add(Cnt[i], Cnt[j]);
    for (int i = 1; i <= N; i++)
        Num[i] = Add(Add(Num[i - 1], Num[i - 1]), Pow(2, i - 1));
    int Ans = 0;
    for (int i = 2; i <= M; i++) {
        int cur = 0;
        for (int j = 1; j <= M / i; j++)
            cur = Add(cur, Mul(Mu[j], Num[Cnt[i * j]]));
        Ans = Add(Ans, Mul(i, cur));
    }
    printf("%d", Ans);
    return 0;
}

ixRic

关注

1
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
[CodeForces 839D] Winter is here（莫比乌斯反演） | 错题本

文章目录题目分析代码题目[CodeForces 839D] Winter is here分析设 mmm 是最大的数，枚举 gcd⁡=d\gcd = dgcd=d，子序列长度为 lll，要求的就是 ∑d=2m∑l=1nd⋅l∑1≤k1<k2<⋯<kl≤n[gcd⁡i=1laki=d]=∑d=2m∑l=1nd⋅l∑1≤k1<k2<⋯<kl≤n(∏i=1l[d∣aki])[gcd⁡i=1lakid=1]\begin{aligned} &\sum_{d = 2}^
复制链接

扫一扫