P3383 【模板】线性筛素数

---

P3383 【模板】线性筛素数

---

时间限制 2.00s
内存限制 512.00MB

题目背景

本题已更新，从判断素数改为了查询第\(k\)小的素数
提示：如果你使用 cin 来读入，建议使用 std::ios::sync_with_stdio(0) 来加速。

题目描述

如题，给定一个范围\(n\)，有\(q\)个询问，每次输出第\(k\)小的素数。

输入格式

第一行包含两个正整数\(n,q\)，分别表示查询的范围和查询的个数。

接下来\(q\)行每行一个正整数\(k\)，表示查询第\(k\)小的素数。

输出格式

输出\(q\)行，每行一个正整数表示答案。

输入输出样例

输入 #1

100 5
1
2
3
4
5

输出 #1

2
3
5
7
11

说明/提示

【数据范围】
对于\(100\%\)的数据，\(n = 10^7\)，\(1 \le q \le 10^6\)，保证查询的素数不大于\(n\)。

Data by NaCly_Fish.

推荐学委的欧拉筛筛素数

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e8+5;
int Q[N];
bool vis[N];
void init(int n){
	for(int i=2;i<=n;++i){
		if(!vis[i]) Q[++Q[0]]=i;
		for(int j=1;Q[j]*i<=n&&j<=Q[0];++j){
			vis[Q[j]*i]=1;
			if(!i%Q[j]) break;
		}
	}
}
int main(){
	int n,q;
	scanf("%d %d",&n,&q);
	init(n);
	while(q--){
		int k;
		scanf("%d",&k);
		printf("%d\n",Q[k]);
	}
	return 0;
}