SP4191 MSKYCODE - Sky Code

题目大意

给出长为 nn 的序列 aa,现在让你求出 aa 中有多少个四元组,满足这四个数的最大公约数等于 11

对于 100%100\% 的数据,满足 1n104,1ai1041 \leq n \leq 10^4,1 \leq a_i \leq 10^4

解题思路

显然是莫比乌斯反演。

我们令 f(n)f(n) 表示四个数最大公因数为 nn 的方案数,F(n)F(n) 表示四个数的 gcd\gcdnn 的倍数的方案数。

显然 f(x)=xdμ(dx)F(d)f(x)=\sum\limits_{x|d}\mu(\frac{d}{x})F(d)

F(x)F(x) 怎么求呢?可以先记录每个数 ii 出现的次数 cnticnt_i,则是 xx 的倍数的数有 xdcntd\sum\limits_{x|d}cnt_d 个,记为 pp,那么 F(x)=(p4)F(x)=\begin{pmatrix}p\\4\end{pmatrix},也就是从 pp 个数中选 44 个数的方案数。

显然,ans=f(1)=1dμ(d)F(d)ans=f(1)=\sum\limits_{1|d} \mu(d)F(d),循坏累加即可。

CODE

#include <bits/stdc++.h>

#define int long long
#define mem(a, b) memset(a, b, sizeof(a))

#define p_b push_back
#define m_p make_pair

const int INF = 0x3f3f3f3f, _ = 1e4;

using namespace std;

bool vis[_ + 7];

int num, prime[_ + 7], mu[_ + 7];

int n, a[_ + 7];

void init()
{
	mu[1] = 1;
	for (int i = 2; i <= _; ++i)
	{
		if (!vis[i])
			prime[++num] = i, mu[i] = -1;
		for (int j = 1; j <= num; ++j)
		{
			if (prime[j] * i > _)
				break;
			vis[prime[j] * i] = 1;
			if (i % prime[j] == 0)
			{
				mu[prime[j] * i] = 0;
				break;
			}
			mu[prime[j] * i] = -mu[i];
		}
	}
}

int calc(int n) { return n * (n - 1) * (n - 2) * (n - 3) / 24ll; }

int F[_ + 7], d[_ + 7];

signed main()
{
	init();
	while (~scanf("%d", &n))
	{
		if (n < 4)
		{
			puts("0");
			continue;
		}
		memset(d, 0, sizeof d);
		memset(F, 0, sizeof F);
		for (int i = 1; i <= n; ++i)
			scanf("%d", &a[i]), d[a[i]]++;
		int ans = 0;
		for (int i = 1; i <= _; ++i)
			for (int j = i; j <= _; j += i)
				F[i] += d[j];
		for (int i = 1; i <= _; ++i)
			if (F[i] >= 4)
				ans += calc(F[i]) * mu[i];
		printf("%lld\n", ans);
	}
	return 0;
}
posted @ 2021-11-29 18:52  蒟蒻orz  阅读(10)  评论(0)    收藏  举报  来源