题目描述:
如题,给定一个范围 n,有 q 个询问,每次输出第 k 小的素数。
输入格式:
第一行包含两个正整数 n, q,分别表示查询的范围和查询的个数。
接下来 q 行每行一个正整数 k,表示查询第 k 小的素数。
输出格式:
输出 q 行,每行一个正整数表示答案。
代码与分析:
#include<bits/stdc++.h>
using namespace std;
const int N = 1e8 + 10;
int n, q, k, cnt, isPrime[N], Prime[N];
void GetPrime() {
//初始化,令每个数均为素数
memset(isPrime, 1, sizeof(isPrime));
isPrime[1] = 0;
for(int i = 2; i <= n; ++i) {
//若这个数没有被筛掉,则记为素数
if(isPrime[i])
Prime[++cnt] = i;
//不管i是不是素数,需要筛掉i和所有已知的素数的积,但是每个积只需要筛一次
//由这个积除以最小质因数的商来筛掉
for(int j = 1; j <= cnt && i * Prime[j] <= n; ++j) {
//若有某个数可以整除i和已经筛选出来的素数,则标记
isPrime[i * Prime[j]] = 0;
//如果当前的i有Prime[j]这个质因数,假设i = k * Prime[j]
//后续就没有必要使用i和更大的质因数继续筛选下去了
//eg.数i * Prime[j + 1] = k * Prime[j] * Prime[j + 1]
//此时应该使用k * Prime[j + 1]进行筛选(除以最小质因数的商)
if(i % Prime[j] == 0)
break;
}
}
}
int main() {
scanf("%d %d", &n, &q);
GetPrime();
while(q--) {
scanf("%d", &k);
printf("%d\n", Prime[k]);
}
return 0;
}