1.最朴素的判断素数-试除法
using namespace std;
int n;
bool is_prime(int n)
{
if(n<2)
return false;
for(int i=2;i<n;i++)
if(n%i==0)
return false;
return true;
}
int main ()
{
while(cin>>n)
{
if(is_prime(n))
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
}
时间复杂度O(n)。
对代码进行优化:
using namespace std;
int n;
bool is_prime(int n)
{
if(n<2)
return false;
for(int i=2;i<=n/i;i++)
if(n%i==0)
return false;
return true;
}
int main ()
{
while(cin>>n)
{
if(is_prime(n))
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
}
遍历约数时遍历到i<=n/i即可,时间复杂度O(√n)。
不建议写成i<=sqrt(n),调用浪费时间;不建议写成i*i<=n,可能出现数值溢出。
2.埃氏筛
#include<cstring>
using namespace std;
const int maxn=1e6;
bool book[maxn]={false};
void Init()
{
book[0]=1;
book[1]=1;
for(int i=2;i<=maxn;i++)
if(book[i]==false)
for(int j=i+i;j<=maxn;j+=i)
book[j]=true;
}
int main ()
{
int n;
cin>>n;
Init();
for(int i=0;i<n;i++)
{
int x;
cin>>x;
if(book[x]==0)
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
}
时间复杂度O(nlogn)
可以进行优化,只用质数去筛选,优化后时间复杂度O(nloglogn)。
原理是从2开始,将每个素数的各个倍数,标记成合数,原理简单较为高效,但是会出现重复标记的问题,比如标记2的倍数时会标记4,6,在标记3的倍数时还会标记一次6。
3.欧拉筛
#include<cstring>
using namespace std;
const int maxn=1e7+10;
bool book[maxn]={false};
int prime[maxn];
int cnt;
void Init()
{
cnt=0;
book[0]=1;
book[1]=1;
for(int i=2;i<=maxn;i++)
{
if(book[i]==0)
{
prime[cnt++]=i;
}
for(int j=0;j<cnt&&prime[j]*i<=maxn;j++)
{
book[i*prime[j]]=1;
if(i%prime[j]==0)
break;
}
}
}
int main ()
{
int n;
cin>>n;
Init();
for(int i=0;i<n;i++)
{
int x;
cin>>x;
if(book[x]==0)
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
}
线性筛,时间复杂度O(n)。
在埃氏筛法的基础上,让每个合数只被它的最小质因子筛选一次,以达到不重复的目的。