并查集——集合(洛谷 P1621)

题目选自洛谷P1621

一道并查集类型的算法题，基本框架不变，就是合并的时候怎么合并，合并哪些数？

筛出素数之后找出在 ‘p~b’ 范围内的素数

然后枚举a~b之间的数，合并他们

最后看看那些数的祖先是自己就好了

题目描述

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

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

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

输入格式

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

输出格式

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

输入输出样例

输入 1

10 20 3

输出 1

7

说明/提示

样例 1 解释

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

数据规模与约定

• 对于 80\%80% 的数据，1≤a≤b≤10^3。
• 对于 100%100 的数据，1≤a≤b≤10^5,2≤p≤b。

 解题代码： 

#include<stdio.h>
#include<iostream>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
using namespace std;
int a,b,p,f[100001],ans;
bool vis[100001];
void init(){
	for(int i=1;i<=b;i++) f[i]=i;
}
int find(int x){
	if(f[x]==x) return x;
	return f[x]=find(f[x]);
}
void change(){
	memset(vis,true,sizeof(vis));
	for(int i=2;i<=b;i++){
		if(vis[i]){
			for(int j=i*2;j<=b;j+=i){
				vis[j]=false;
				if(i>=p){
					int xx=find(i); int yy=find(j);
					if(xx!=yy) f[xx] = yy;
				}
			}
		}
	}
}
int main(){
	cin>>a>>b>>p;
	init();
	change();
	for(int i=a;i<=b;i++)
		if(i==f[i]) ans++;
	cout<<ans;
	return 0;
}