玲珑杯热身赛A--Alarm （找规律）

#include <cstdio>
#include <cstring>
#include <iostream>
#include <cmath>
#include<algorithm>
using namespace std;

const int maxn=1000007;
const int INF=0x3f3f3f3f;

long long v[maxn], a[maxn], k;
void Isprime()
{
    memset(v, 0, sizeof(v));
    for(int i=2; i<maxn; i++)
    {
        if(v[i]==0)
        {
            a[k++]=i;

            for(int j=i+i; j<maxn; j+=i)
                v[j]=1;
        }
    }
}
int main()
{
    k=1;
    Isprime();

    int T;
    scanf("%d", &T);

    while(T--)
    {
        int n;
        scanf("%d", &n);
        printf("%lld\n", a[n]*a[n]-n);
    }
    return 0;
}