洛谷P3383 【模板】线性筛素数

【模板】线性筛素数

题目背景

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

题目描述

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

100 5
1
2
3
4
5

样例输出 #1

2
3
5
7
11

提示

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

Data by NaCly_Fish.

这道题是多组样例输入，那么复杂度就很重要。普通的素数筛肯定不行了，所以要用到欧拉筛或埃氏筛。
只要你会以上两种筛法的一种，再加一个多组输入加查询就行了。还要注意他要求是第k小所以，加个计数器进行判断，就行了。

Code

#include <bits/stdc++.h>
using namespace std;
bool isPrime[100000010];
int Prime[6000010], cnt = 0;
void GetPrime(int n){
	memset(isPrime, 1, sizeof(isPrime));
	isPrime[1] = 0;
	for(int i = 2; i <= n; i++){
		if(isPrime[i]){
			Prime[++cnt] = i;
		}	
		for(int j = 1; j <= cnt && i*Prime[j] <= n;j++) {
			isPrime[i*Prime[j]] = 0;
			if(i % Prime[j] == 0){
				break;
			}
		}
	}
}
int main(){
	int n, q;
	cin >> n >> q;
	GetPrime(n);
	while (q--){
		int k;
		cin >> k;
		cout << Prime[k] << endl;
	}
	return 0;
}