P3383 【模板】线性筛素数
题目描述
如题,给定一个范围N,你需要处理M个某数字是否为质数的询问(每个数字均在范围1-N内)
输入格式
第一行包含两个正整数N、M,分别表示查询的范围和查询的个数。
接下来M行每行包含一个不小于1且不大于N的整数,即询问该数是否为质数。
输出格式
输出包含M行,每行为Yes或No,即依次为每一个询问的结果。
输入输出样例
输入
100 5
2
3
4
91
97
输出
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。
AC代码
#include <stdio.h>
#include <string.h>
bool book[10000005];
int prime[10000005], p = 0;
int isPrime(int n)
{
memset(book,true,sizeof(book));
book[0] = false;
book[1] = false;
for(int i = 2; i <= n; ++ i){
if(book[i]){
prime[p ++] = i;
}
for(int j = 0; j < p && prime[j] * i <= n; ++ j){
book[prime[j] * i] = false;
if(!i % prime[j]){
break;
}
}
}
return book[10000005];
}
int main(void)
{
int n, m, a;
int isPrime(int n);
scanf("%d%d",&n,&m);
isPrime(n);
while(m --){
scanf("%d",&a);
if(book[a] == true) printf("Yes\n");
if(book[a] == false) printf("No\n");
}
return 0;
}
批注
注意这里关于素数筛(欧式筛)的使用(函数部分)
注意bool类型在C++当中才有,C中没有
注意memset函数的头文件为<string.h>
来源:CSDN
作者:Kapo1
链接:https://blog.csdn.net/Kapo1/article/details/103515613