【知识点4】素数(质数)

1. 素数的概念

素数又称质数，是指除了1和本身之外，不能被其他数整除的一类数，反之，则称为合数。

注意：小于等于1的数既不是素数，也不是合数。

本节将解决两个问题：

1. 如何判断给定的正整数n是否是质数；
2. 如何在较短的时间内得到1~n内的素数表。

2. 素数的判断

解法一：暴力。由素数的定义可以遍历从2 ~ n-1的数，判断它们是否可以整除n。但是这样时间复杂度就到达了 $O(n)$，通常判断素数只会作为程序的一部分，所以这样的时间复杂度还是有点高。

解法二：开根号。如果在2 ~ n-1中存在n的约数，不妨设为k，即n % k == 0，那么由 k * （n / k）== n可知，n/k也是n的一个约数，且k与n/k中一定满足其中一个小于等于$\sqrt{n}$，一个大于等于$\sqrt{n}$。这样我们只要判断2 ~ $\sqrt{n}$之间的整数就好了，时间复杂度为 $O(\sqrt{n})$。

代码如下：

bool isPrime(int n){
	if(n <= 1){
		return false;	//特判 
	}
	int sqr = (int) sqrt(1.0 * n);	//根号n,这个1.0要乘，（int）也要加 
	for(int i=2;i <= sqr;++i){
		if(n % i == 0){	//注意是取余，而不是除法
			return false;
		}
	}
	return true;	//否则，就是true 
} 

注意：

1. 上述代码中，sqrt()的作用为一个浮点数开根号，需要添加math.h头文件。由于sqrt的参数要求是浮点数，因此在n前面乘以1.0来使其变成浮点数。
2. isPrime()一定要记得特判小于等于$1$不是质数！！！

3. 素数表的获得

3.1 暴力

通过上面的学习，我们知道了怎么判断一个数n是否是素数，那么我们如果要统计1 ~ n之间的素数也就清楚了，算法的时间复杂度为$O(n\sqrt{n})$，当我们的数据n小于$10^5$时，这种算法是不会超时的。代码如下：

const int maxn = 101;	//步长
int prime[maxn],pNum = 0;	//prime数组存放所有的素数，pNum用于存储素数数量
bool p[maxn] = {0};	//p[i] == true表示i是素数

void Find_Prime(){
	for(int i = 1;i < maxn;++i){	//不能写成i <= maxn 
		if(isPrime(i) == true){
			prime[pNum++] = i;
			p[i] = true;
		}
	}
}

数据结构：

1. int prime[]用于存储所有的素数；
2. bool p[]用于快速判断某一个数是否为素数；
3. pNum用于记录素数的数量。

3.2 性能优化——筛法

我们使用下面的“筛法”，达到时间复杂度为 $O(nloglogn)$。

"筛"的实现，可以使用一个bool型数组p来标记，如果a被筛掉，那么设置p[a]为true；否则，p[a]为false。程序初始化时，记得初始化p数组全为false。
代码如下：

const int maxn = 101;
int prime[maxn],pNum = 0;
bool p[maxn] = {0};
void Find_Prime(){
	for(int i = 2;i<maxn;i++){	//要从2开始，i<maxn结束！！！ 
		if(p[i] == false){	//如果i是素数 
			prime[pNum++] = i;	//把素数i存放到prime数组中 
			for(int j = i+i;j < maxn;j += i){
				//筛掉所有i的倍数，循环条件不能写成j<=maxn
				p[j] = true; 
			} 
		}
	}
}

4. 注意点

注意：

1. 上述代码中，sqrt()的作用为一个浮点数开根号，需要添加math.h头文件。由于sqrt的参数要求是浮点数，因此在n前面乘以1.0来使其变成浮点数。
2. isPrime()一定要记得特判小于等于$1$不是质数！！！

注意上面的条件非常重要！！！虽然筛法简单，但是做题目的时候很容易出错

5. 题型训练

1. 【PAT B1007】素数对猜想
2. 【PAT B1013】数素数
3. 【PAT A1015】Reversible Primes
4. 【PAT A1078】Hashing
5. 素数判定——哈工大
6. 素数——北航
7. LeetCode 204. Count Primes
8. PAT A1116 Come on! Let’s C
9. ⭐PAT A1152 Google Recruitment

6. 参考资料

1. 算法笔记