欧拉筛
题目
提示:本题输入输出、运算数据量较大。
- 对于 C++ 语言,如果你使用
cin来输入输出,建议使用std::ios::sync_with_stdio(0)来加速,同时使用'\\\\n'换行输出。 - 对于 Java 语言,使用线性筛并且优化输入输出,也可以在规定时限内通过本题,但是时限可能较紧张。
- 对于 Python 语言,语言性能差异较大,需要使用到
numpy库的数组以替代列表,且使用埃氏筛法,依然可以在合适的时间和内存消耗下通过本题。
题目描述
如题,给定一个范围 n,有 q个询问,每次输出第 k小的素数。
输入格式
第一行包含两个正整数 n,q,分别表示查询的范围和查询的个数。
接下来 q 行每行一个正整数 k,表示查询第 k小的素数。
输出格式
输出 q 行,每行一个正整数表示答案。
输入输出样例 #1
输入 #1
输出 #1
说明/提示
【数据范围】
对于 $100\%$ 的数据,$n = 10^8$,$1 \le q \le 10^6$,保证查询的素数不大于 $n$。
思路
质数查找。
本题要求查找特定范围内的质数,快速查找可以用欧拉筛或埃氏筛,考虑到欧拉筛可以在一轮循环中对已有质数进行统计,因此选择了欧拉筛。
和埃氏筛相同,欧拉筛的核心在于一个数的任意整数倍数一定是合数,而欧拉筛减少了对于同一个合数的重复计算(如,埃氏筛中会对16筛两次:28和44),其思路如下:
- 从2开始遍历,如果当前数在合数数组中为false,就计入质数数组;
- 用当前数去乘质数数组中的每一个数,并将结果分别标记为合数;
- 特别的,如果当前数是某一个质数的倍数,就跳出乘算。
Code
#include<iostream>
using namespace std;
const int N=1e8+1;
const int M=1e6+5;
int n,q,d;
int pre[N]={0};
bool comp[N]={false};
int index=0;
int main(){
std::ios::sync_with_stdio(0);
cin>>n>>q;
for(int i=2;i<N;i++){
if(comp[i]==false){
pre[index]=i;
index++;
}
for(int j=0;j<index;j++){
if(i*pre[j]>n)break;
comp[i*pre[j]]=true;
if(i%pre[j]==0)break;
}
}
for(int i=1;i<=q;i++){
cin>>d;
cout<<pre[d-1]<<"\\n";
}
return 0;
}