洛谷：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.

线性筛素数模板，不熟可以多加练习

#include <bits/stdc++.h>
// #include<iostream>
// #include<iomanip>
// #include<vector>
// #include<queue>
// #include<cstring>
// #include <algorithm>
// #define int long long
#define rep(i, l, r) for (int i = l; i <= r; i++)
using namespace std;

#define endl '\n'
const int M = 1e7 + 7;
const int N = 1e8 + 7;

int primes[M],cnt=0;
bool st[N];
void getprime(int n)
{
  for(int i=2;i<=n;i++)
  {
    if(!st[i])
    {
         primes[cnt++]=i;
    }
    for(int j=0;primes[j]<=n/i;j++)
    {
      st[primes[j]*i]=1;
      if(i%primes[j]==0)break;
    }
  }

}

int main()
{
  std::ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
 int n,t;
cin>>n>>t;
 getprime(n);
   while(t--)
   {
    int m;
    cin>>m;
   cout<<primes[m-1]<<endl;
   }
  return 0;
}