集合

P1621 集合

https://www.luogu.com.cn/problem/P1621
洛谷题
题目描述

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

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

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

输入格式

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

输出格式

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

输入输出样例

输入 #1 复制
10 20 3
输出 #1 复制
7
该题目的意思很明确，先用素数筛找出p到b范围内素数，再用枚举法找出符合要求的数，最后将它们合并，找出子集数，最后输出就可以了。

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN = 1e5+20;
int S[MAXN];
bool prime[MAXN];
int Find(int x) {
    return S[x] ==x ? x : (S[x] = Find(S[x]));//寻找x的父节点
}
void Union(int x1,int x2) {//把x1和x2节点或集合合并。
    int rt1 = Find(x1);
    int rt2 = Find(x2);
    if(rt1!=rt2) S[rt2] = rt1;
}
int a, b, p;
void make_prime(void) {//标记1到b之间的素数
    memset(prime,1,sizeof(prime));
    prime[0]=prime[1]=0;
    for(int i=2; i<=b; i++) 
    if(prime[i]) 
    for(int j=2*i; j<=b; j+=i) 
    prime[j] = 0;
}
int rec[MAXN], cnt;
int main() {
    cin >> a >> b >> p;
    for (int i = a; i <= b; i++) S[i]=i;//集合初始化
    make_prime();
    for (int i = p; i <= b; i++) 
    if(prime[i]) rec[cnt++] = i;//存储p到q自己的素数
    for (int i = 0; i < cnt; i++) {
        int k = 0;
        while(k*rec[i]<a) k++;
        while((k+1)*rec[i]<=b) {
            Union(k * rec[i], (k + 1) * rec[i]);//将用相同质因数的合并
            k++;
        }
    }
    int res = 0;
    for (int i = a; i <= b; i++) if(S[i]==i) res++;//查找集合个数
    cout << res << endl;
    return 0;
}

如有错误，欢迎指正。