【洛谷】P3383 【模板】线性筛素数(埃拉托斯特尼筛法) 题解
原题地址:https://www.luogu.org/problem/P3383
题目描述
如题,给定一个范围N,你需要处理M个某数字是否为质数的询问(每个数字均在范围1-N内)
输入输出格式
输入格式:
第一行包含两个正整数N、M,分别表示查询的范围和查询的个数。
接下来M行每行包含一个不小于1且不大于N的整数,即询问该数是否为质数。
输出格式:
输出包含M行,每行为Yes或No,即依次为每一个询问的结果。
输入输出样例
输入样例#1:
100 5
2
3
4
91
97
输出样例#1:
Yes
Yes
No
No
Yes
说明
时空限制:500ms 128M
数据规模:
对于30%的数据:N<=10000,M<=10000
对于100%的数据:N<=10000000,M<=100000
样例说明:
N=100,说明接下来的询问数均不大于100且不小于1。
所以2、3、97为质数,4、91非质数。
故依次输出Yes、Yes、No、No、Yes。
思路:
1、这道题不能用我们平时的求素数方法,即直接遍历n个数,对每个数再遍历到sqrt(n),时间复杂度是O(n^(3/2)),时间限制是500ms,肯定超时。
2、这里用的是一般筛法(埃拉托斯特尼筛法),基本思想:素数的倍数一定不是素数。
3、洛谷对输入输出的要求不高,这道题的m个数可以输入一个就输出一个,亦可以等全部输入完再一起输出,在这里我采用的是第二种方法。
4、用数组a[i]保存输入的m个数,数组b[i]用来判断是否为素数,0表示为不是素数,1表示是素数,先全部初始化为1。0和1均不是素数,然后从2开始,把2的倍数均标记为0,即不是素数,一直到大于n,然后从下一个素数3开始,进行同样的处理,到最后,数组中为1的即为素数。
代码如下:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=1e7;
//n表示范围,m表示个数,数组a[i]表示输入的数、数组b[i],0表示不是为素数,1表示是素数
int n,m,x,a[N],b[N];
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++) //输入m个数
scanf("%d",&a[i]);
memset(b,1,sizeof(b)); //初始化,把所有的数标记为是素数
b[0]=b[1]=0; //0和1不是素数
for(int i=2;i<=n;i++) //从2开始遍历
{
if(b[i]) //如果标记为素数,即未改变初始化的值
{
for(int j=i+i;j<=n;j+=i) //把所有该数的倍数标记为0,即不是素数
b[j]=0;
}
}
for(int i=1;i<=m;i++)
{
if(b[a[i]]) //如果输入的数是素数,输出Yes
printf("Yes\n");
else //否则输出No
printf("No\n");
}
return 0;
}