题目链接:
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;
}
如果我的文章对你有所帮助,不妨给我个关注何点赞吧。