玲珑学院 1010 - Alarm

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <cmath>
#include <vector>
#include <algorithm>
using namespace std;
#define lowbit(x) (x&(-x))
#define max(x,y) (x>y?x:y)
#define min(x,y) (x<y?x:y)
#define mem(a) (memset(a,0,sizeof(a)))
typedef long long ll;
ll a[105000];
ll vis[10006];
void getprime()
{
    memset(a,0,sizeof(a));
    a[1]=1;
    for(ll i=2;i<105000;i++)
    {
        if(a[i]==1) continue;
        for(ll j=2;j*i<105000;j++)
            a[j*i]=1;
    }
    ll ans=1;
    for(ll i=2;i<105000;i++)
    {
        if(!a[i]) vis[ans++]=i;
        if(ans>=10002) break;
    }
}
int main()
{
    ll t,n;
    scanf("%lld",&t);
    getprime();
    while(t--)
    {
        scanf("%lld",&n);
        printf("%lld\n", vis[n]*vis[n]-n);
    }
}