洛谷 P1621 集合

题目链接：

https://www.luogu.com.cn/problem/P1621https://www.luogu.com.cn/problem/P1621

---

题目：

Caima 给你了所有 [a,b] 范围内的整数。一开始每个整数都属于各自的集合。每次你需要选择两个属于不同集合的整数，如果这两个整数拥有大于等于 p 的公共质因数，那么把它们所在的集合合并。

重复如上操作，直到没有可以合并的集合为止。

现在 Caima 想知道，最后有多少个集合。

输入格式：

一行，共三个整数 a,b,p，用空格隔开。

输出格式：

一个数，表示最终集合的个数。

例：

输入：

10 20 3

输出：

7

样例解释：

对于样例给定的数据，最后有 {10,20,12,15,18} , {13} , {14} , {16} , {17} , {19} , {11}  共 77 个集合，所以输出应该为 7。

数据提示：

• 对于 100% 的数据，1 ≤ a ≤ b ≤  , 2 ≤ p ≤ b 。

---

思路：

首先看题，可知要讲数字放入同一个集合中。那么我们应该反应过来这道题目考察的是并查集。我们就可以拿出并查集模板了

class Union {
	public:
		Union(int n) :fa(n + 1) {
			for (int i = 0; i < n; i++) {
				fa[i] = i;
			}
		}

		int find(int i) {
			return fa[i] = (fa[i] == i ? i : find(fa[i]));
		}

		void merge(int x, int y) {
			if (find(x) == find(y)) {
				return;
			}
			fa[find(x)] = find(y);
			return;
		}

		vector<int> fa;
};

因为数据范围可能很大，所以我们使用埃氏筛来筛查素数。

(不懂埃氏筛的可以看这里 -> 一次找出范围内的所有素数，埃式筛法是什么神仙算法？ - 知乎 (zhihu.com)https://zhuanlan.zhihu.com/p/146418699)

在筛查的过程中，因为集合数量最多不会超过(b-a+1)所以让最终结果ans等于(b-a+1)，当前数字i≥p并且i为素数时，让i成倍增长，让其倍数与i相连通，并且让ans减一。

最后输出ans即可。

---

AC代码：

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<vector>

using namespace std;

class Union {
	public:
		Union(int n) :fa(n + 1) {
			for (int i = 0; i < n; i++) {
				fa[i] = i;
			}
		}
		int find(int x) {
			return fa[x] = (fa[x] == x ? x : find(fa[x]));
		}

		void merge(int x, int y) {
			if (find(x) == find(y)) {
				return;
			}
			fa[find(x)] = find(y);
			return;
		}

		vector<int> fa;
};

int a, b, p;
bool arr[100005] = { 0 };

int main() {

	scanf("%d%d%d", &a, &b, &p);
	Union u(b);
	int ans = b - a + 1;
	//将全部素数(除了0和1)变为假
	for (int i = 2; i <= b; i++) {
		if (!arr[i]) {
			if (i >= p) {
				for (int j = i * 2; j <= b; j += i) {
					arr[j] = 1;
					if ((j - i) >= a && u.find(j) != u.find(j - i)) {
						u.merge(j, j - i);
						ans--;
					}
				}
			}
			else {
				for (int j = i * 2; j <= b; j += i) {
					arr[j] = 1;
				}
			}
		}
	}
	printf("%d", ans);

	return 0;
}

如果我的文章对你有所帮助，不妨给我个关注何点赞吧。