本文总结自《挑战程序设计竞赛》
关于素数的算法有许多,其中在程序设计竞赛中常用到的是艾氏筛法,简单的素数测试和整数分解这类算法。
1. 素数测试
所谓素数,是指恰好有2个约数的整数。因为n的约数都不超过n,所以只要检查 2~n-1的所有整数是否整除n就能判定n是不是素数。因此,如果d是n的约数,那么n/d也是n的约数,由n=d*n/d可知min(d,n/d)<=√n,所以只要检查2~√n的所有整数就足够了。同理可知,整数分解和约数枚举都可以在O(√n)时间完成。
eg:给定整数n,判断n是否为素数。限制条件: 1 <= n <= 1000000000.
#include <iostream>
#include <vector>
#include <map>
#include <cstdio>
using namespace std;
//素数测试
bool is_prime(int n){
for(int i = 2;i*i<=n;i++){
if(n % i == 0){
return false;
}
}
return n!=1; //1是例外
}
//约数枚举
vector<int> divisor(int n){
vector<int> res;
for(int i = 1;i*i <= n;i++){
if(n%i == 0){
res.push_back(i);
if(i != n/i) res.push_back(n/i); //防止重复的数输出两遍,比如9=3*3,但是我们只需要输出一个3
}
}
return res;
}
//整数分解
map<int,int> prime_factor(int n){
map<int,int> res;
for(int i = 2;i*i<=n;i++){
while(n % i == 0){
++res[i];
n = n/i;
}
}
if(n != 1) res[n] = 1;
return res;
}
int main(){
int t,n;
vector<int> res1;
map<int,int> res2;
cin>>t;
while(t -- != 0){
cin>>n;
res1 = divisor(n);
res2 = prime_factor(n);
cout<<"素数测试结果:"<<is_prime(n)<<endl;
cout<<"约数枚举: "<<endl;
for(int i = 0;i<res1.size();i++){
cout<<res1[i]<<" ";
}
cout<<endl;
cout<<"整数分解:"<<endl;
for(int i = 0;i<res2.size();i++){
cout<<res2[i]<<" ";
}
cout<<endl<<"----------------------------------"<<endl;
}
return 0;
}
2. 艾氏筛法
如果只对一个整数进行素数测试,通常O(√n)的算法就足够了。但如果要对许多整数进行素数测试,则有更为高效的算法。 首先,将2到n范围内的所有整数写下来,其中最小的数字2是素数。将表中所有2的倍数都划去。表中剩余的数字将会是3,它不能被更小的数整除所以是素数,再将表中
从所有3的倍数都划去。以此类推,如果表中剩余的最小数字是m时,m就是素数。然后将表中所有m的倍数都划去。像这样反复操作,就能依次枚举n以内的素数。 此外,艾氏筛法的复杂度仅有O(nloglogn)。对于程序设计竞赛中的数据规模,讲它的复杂度看作大致是线性的也无妨。
2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 | 16 | 17 | 18 | 19 | 20 |
2 | 3 | - | 5 | - | 7 | - | 9 | - | 11 | - | 13 | - | 15 | - | 17 | - | 19 | - |
2 | 3 | - | 5 | - | 7 | - | - | - | 11 | - | 13 | - | - | - | 17 | - | 19 | - |
eg:素数的个数 》》 给定整数n,请问n以内有多少个素数?
#include <iostream>
#include <cstdio>
#define MAX_N 10000005
using namespace std;
int prime[MAX_N]; //第i个素数
bool is_prime[MAX_N+1]; //is_prime[i]为true表示i是素数
int sieve(int n){
int p = 0;
for(int i = 0;i<=n;i++){
is_prime[i] = true;
}
is_prime[0] = is_prime[1] = false;
for(int i = 2;i<=n;i++){
if(is_prime[i]){
prime[p++] = i;
for(int j = 2*i;j<=n;j+=i){
is_prime[j] = false;
}
}
}
return p;
}
int main()
{
int t,n;
cin>>t;
while(t-- != 0){
cin>>n;
cout<<n<<"以内的素数个数为:"<<sieve(n)<<endl<<"-------------------"<<endl;
}
}
3. 区间筛法
参考《挑战程序设计竞赛》P120。