【题解】CF1780F-Three Chairs

最新推荐文章于 2024-10-01 04:50:10 发布

inferior_hjx

最新推荐文章于 2024-10-01 04:50:10 发布

阅读量90

点赞数 4

分类专栏：题解文章标签：算法 c++

本文链接：https://blog.csdn.net/inferior_hjx/article/details/129633560

版权

题解专栏收录该内容

32 篇文章 0 订阅

订阅专栏

题目(cf)

题目(luogu)

题目大意

给定长度为 $n(3\le n\le 3\times 10^5)$ 的数组 $a(1\le a_i\le 3\times 10^5)$ ，其中的元素互不相同。问有多少三元组满足最大数与最小数互质。两个三元组不同当且仅当存在数 $x$ 出现在三元组 $A$ 中而不出现在三元组 $B$ 中。

题解

容易发现数组 $a$ 的顺序对答案没有影响。为了方便统计答案，我们人为地给它安排顺序，也就是对 $a$ 数组排序。

显然，当确定了三元组中最大值和最小值后，中间值可以随意选，只需要满足它位于最大最小值之间。也就是说，当最大值为 $a_i$ 而最小值为 $a_j$ （显然 $j < i$ ）时，中间值有 $i - j - 1$ 种取值。

易得

$ans=\sum\limits_{gcd(a_i,a_j)=1,j<i}(i-j-1)$

对于每个 $i$ 考虑其贡献，有

$ans=\sum\limits_{i=1}^{n}\sum\limits_{gcd(a_i,a_j)}(i - j - 1)\\ =\sum\limits_{i=1}^{n}\sum\limits_{j=1}^{i-1}(i - j - 1)[gcd(a_i, a_j)=1]$

以下是莫反技巧推导。

$ans=\sum\limits_{i=1}^{n}\sum\limits_{j=1}^{i-1}(i - j - 1)[gcd(a_i, a_j)=1]\\ =\sum\limits_{i=1}^{n}\sum\limits_{j=1}^{i-1}(i - j - 1)\sum\limits_{k|gcd(a_i, a_j)}\mu(k)$

考虑将 $\mu$ 提到最前面，变换枚举对象，枚举 $d=gcd(a_i,a_j)$ 。

$ans=\sum\limits_{d=1}^{a_n}\mu(d)\sum\limits_{d|a_i}\sum\limits_{d|a_j,j<i}(i-j-1)$

考虑分离 $i$ 和 $j$ .

$ans=\sum\limits_{d=1}^{a_n}\mu(d)\sum\limits_{d|a_i}\left(\sum\limits_{d|a_j,j<i}(i-1)-\sum\limits_{d|a_j,j<i}j\right)$

整理出容易解决的式子：

$ans=\sum\limits_{d=1}^{a_n}\mu(d)\sum\limits_{d|a_i}\left((i-1)\sum\limits_{j=1}^{i-1}[d|a_j]-\sum\limits_{d|a_j,j<i}j\right)$

$\mu(d)$ 可以欧拉筛线性预处理。

对于 $d$ 和 $a_i$ ，可以直接枚举，时间复杂度为 $O(n+\dfrac n 2+\dfrac n 3+\cdots+\dfrac n n)=O(n\times (1+\dfrac 1 2+\dfrac 1 3+\cdots+\dfrac n n))=O(n\ln n)$ .

关于 $j$ 的项，可以在枚举 $i$ 的过程中累加统计，因为 $j$ 必定小于 $i$ ，也就是说之前其实已经遍历过 $j$ 了。具体见代码。

总的时间复杂度是 $O(n\ln n)$ .

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 300005;
int n, a[N], mu[N], pri[N], cnt = 0, vis[N], b[N];
LL ans = 0;
void init() {
    mu[1] = 1;
    for (int i = 2; i < N; i++) {
        if (!vis[i]) pri[++cnt] = i, mu[i] = -1;
        for (int j = 1; j <= cnt && i * pri[j] < N; j++) {
            vis[i * pri[j]] = 1;
            if (i % pri[j] == 0) break;
            mu[i * pri[j]] = -mu[i];
        }
    }
}
int main() {
    init();//莫比乌斯函数线性筛
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
    sort(a + 1, a + n + 1);
    for (int i = 1; i <= n; i++) b[a[i]] = i;//b[i]表示值为i的元素在数组中出现的位置
    for (int d = 1; d <= a[n]; d++) {//枚举d
        LL sum = 0, s1 = 0, s2 = 0;//sum统计后面的一大串式子，s1统计前i-1个数有多少出现在数组中且整除d，s2统计出现在数组中且整除d的元素的位置（下标）和
        for (int i = d; i <= a[n]; i += d)//枚举值，值在数组中的位置是b[i]
            if (b[i])//数组中有i这个数
                sum += 1ll * (b[i] - 1) * s1 - s2, s1++, s2 += b[i];//维护
        ans += sum * mu[d];
    }
    printf("%lld", ans);
    return 0;
}