【luogu P2714】四元组统计（高维前缀和 / 差分）

最新推荐文章于 2023-10-03 13:47:22 发布

SSL_TJH

最新推荐文章于 2023-10-03 13:47:22 发布

阅读量191

点赞数

分类专栏： # 数学或数论文章标签：高维前缀和高维差分

要转发的话记得附上链接哦~（不过也不会有人转发的啦）

本文链接：https://blog.csdn.net/weixin_43346722/article/details/125105629

版权

数学或数论专栏收录该内容

285 篇文章 1 订阅

订阅专栏

四元组统计

题目链接：luogu P2714

题目大意

给你 n 个数，问你有多少个四元组满足它们四个的 gcd 为 1。

思路

考虑到我们是要 $\gcd$ 为 $1$ ，那就是要每个质数每个数的出现个数的最小值都是 $0$ 。
考虑一个类似容斥的过程，就是我们可以求 $f_i$ 表示有多少个数至少都是 $i$ 的倍数。（这个直接高维前缀和）

那你求出个数 $x$ 之后直接变成 $\Large\binom{x}{4}$ ，然后你这个部分还要容斥，那你可以莫反（所以其实你可以不用高维前缀和而是把莫反 $\mu$ 不是 $0$ 的位置只求这些好像也可以过还很快）。
也可以直接高维差分。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long

using namespace std;

const int N = 1e4 + 100;
int n, a[N], m, prime[N];
bool np[N];
ll f[N];

void Init() {
	for (int i = 2; i < N; i++) {
		if (!np[i]) prime[++prime[0]] = i;
		for (int j = 1; j <= prime[0] && i * prime[j] < N; j++) {
			np[i * prime[j]] = 1; if (i % prime[j] == 0) break;
		}
	}
}

int main() {
	Init();
	
	while (scanf("%d", &n) != EOF) {
		m = 0; for (int i = 1; i <= n; i++) scanf("%d", &a[i]), m = max(m, a[i]);
		for (int i = 1; i <= m; i++) f[i] = 0; for (int i = 1; i <= n; i++) f[a[i]]++;
		
		for (int i = 1; i <= prime[0] && prime[i] <= m; i++)//高维前缀和
			for (int j = m / prime[i]; j >= 1; j--)
				f[j] += f[j * prime[i]];
		for (int i = 1; i <= m; i++) f[i] = f[i] * (f[i] - 1) * (f[i] - 2) * (f[i] - 3) / 4 / 3 / 2;//乘上组合数
		for (int i = 1; i <= prime[0] && prime[i] <= m; i++)//高维差分
			for (int j = 1; j * prime[i] <= m; j++)
				f[j] -= f[j * prime[i]];
		printf("%lld\n", f[1]);
	}
	
	return 0;
}