题目描述
如题,给定一个范围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。
素数筛法:https://blog.csdn.net/zhaohaibo_/article/details/89044700
埃式( O ( N l o g l o g N ) O(NloglogN) O(NloglogN))+二分
# include <iostream>
using namespace std;
const int maxn = 10000010;
int n, m, prime[maxn], tot, s;
bool v[maxn];
void findPrime()
{
for (int i = 2; i <= n; i++)
{
if (v[i]) continue;
prime[tot++] = i;
for (int j = i; j <= n/i; j ++) v[i*j] = 1;
}
}
int main() {
cin >> n >> m;
findPrime();
while(m--) {
cin >> s;
int l = 0, r = tot-1, mid;
while(l < r) {
mid = (l + r) >> 1;
if (s <= prime[mid]) r = mid;
else l = mid + 1;
}
if(s == prime[l]) cout << "Yes" << endl;
else cout << "No" << endl;
}
return 0;
}
欧式( O ( N ) O(N) O(N)) + 二分
#include <iostream>
using namespace std;
const int maxn = 10000010;
int n, m, prime[maxn], tot, s;
int v[maxn];
void findPrime()
{
for (int i = 2; i <= n; i ++)
{
if (v[i]==0) {v[i] = i; prime[tot++] = i;}
for (int j = 0; j < tot; j ++)
{
if (prime[j] > v[i] || prime[j] > n/i) break;
v[i*prime[j]] = prime[j];
}
}
}
int main() {
cin >> n >> m;
findPrime();
while(m--) {
cin >> s;
int l = 0, r = tot-1, mid;
while(l < r) {
mid = (l + r) >> 1;
if (s <= prime[mid]) r = mid;
else l = mid + 1;
}
if(s == prime[l]) cout << "Yes" << endl;
else cout << "No" << endl;
}
return 0;
}