质数相关算法

#include <iostream>
#include <vector>
#include <unordered_map>
#include <algorithm>

using namespace std;
typedef long long LL;

// 试除法判断质数
bool is_prime(int x)
{
	if (x < 2)
	{
		return false;
	}

	for (int i = 2; i <= x / i; i++)
	{
		if (x % i == 0)
		{
			return false;
		}
	}

	return true;
}



// 分解质因数(把一个合数写成几个质数相乘的形式)
void divide(int x)
{
	// 一个合数x 至多只有一个大于 sqrt(x) 的质因子，将此质因子独立讨论
	for (int i = 2; i <= x / i; i++)
	{
		// 能进入此判断语句的 i 均为质数
		if (x % i == 0)
		{
			while (x % i == 0)
			{
				cout << i << " ";
				x /= i;
			}
		}
	}

	// 将大于sqrt(x)的质因子单独讨论
	if (x > 1)
	{
		cout << x << endl;
	}
}




const int N = 10000010;

// 质数数组存储质数
int primes[N], cnt;
bool st[N]; // st[i]为true表示i不是质数，被筛掉了


// 筛选质数（区间 [1,n] 中有多少个质数）
int get_prime(int n)
{
	// 如果过程中会多次调用此函数，则每次调用是要清空质数数组，否则会产生堆积
	cnt = 0;

	for (int i = 2; i <= n; i++)
	{
		// 从2开始，如果该数没有被筛掉的话，则为质数
		if (!st[i])
		{
			primes[cnt] = i; // 加入质数表
			cnt++; // 计数+1


			// 将当前数的所有倍数筛除掉
			for (int j = i + i; j <= n; j += i)
			{
				st[j] = true;
			}
		}
	}
	
	return cnt;
}


// 筛选质数的另一种筛法（线性筛法）,核心思想：每个合数均会被其最小质因子筛去
int get_prime_2(int n)
{
	cnt = 0;

	for (int i = 2; i <= n; i++)
	{
		// 如果没有被筛掉，则为质数
		if (!st[i])
		{
			primes[cnt] = i;
			cnt++;
		}

		// 将已确定的质数的i倍数筛掉,直到枚举到i的最小质因子或者 primes[j] * i > n
		for (int j = 0; primes[j] <= n / i; j++)
		{
			st[primes[j] * i] = true; // 保证每个合数都被它的最小质因子筛除掉

			// p[j]是i的最小质因子，p[j]又是（p[j]*i）的最小质因子
			// 如果不break掉，进入下一轮循环时，p[j+1] > p[j]；那么下一轮中的 (p[j+1]*i) 的最小质因子就不是 p[j+1] 了，从而无法保证每个数都是被自己的最小质因子筛掉的，导致某些数会被重复筛选
			if (i % primes[j] == 0)
			{
				break;
			}
		}
	}

	return cnt;
}