题目是P3383传送门 我的代码 c++

#include<bits/stdc++.h>
using namespace std;
#define RI register int
inline int read(){
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){s=s*10+ch-'0';ch=getchar();}
	return s*w;
}
const int MAXN=10000001;
int n1,prime[MAXN],not_prime[MAXN],tot,q;
void sieve(int n)
{
	for(RI i=2;i<=n;++i)
	{
		if(not_prime[i]==0)
		 prime[tot++]=i;
		for(RI j=0;j!=tot&&i*prime[i]<=n;j++)
		{
			not_prime[i*prime[j]]=1;
			if(i%prime[j]==0)break;
		}
	}
}
int main()
{
	n1=read();q=read();
	int k[q+1];
	sieve(n1);
	for(RI i=1;i<=q;i++)
	k[i]=read();
	for(RI i=1;i<=q;i++)
	printf("%d\n",prime[k[i]-1]);
	return 0;
}