质方数

Problem Description
小明天生对数字比较敏感，3岁的时候就能背诵圆周率一百位。

现在，小明慢慢长大了，但依然很喜欢数字，最近，他迷上了质数和平方数，并且自己把质数的平方命名为“质方数”。
现在，他在研究这样一个问题：距离一个正整数N最接近的质方数是多少？


Input
输入数据第一行是一个正整数T（T<=20），表示有T组输入数据。
接下来T行，每行输入一个正整数N（1<=N<=10^8）。


Output
对于每组数据，请输出距离N最接近的质方数，每组输出占一行。


Sample Input
2
1
10


Sample Output
4

9

#include <iostream>
#include <cmath>
#include <cstdio>
using namespace std;
#define N 10005
long long int T,n,a[N];
int prime(int n)
{
   for(int i=2;i<=sqrt(n);i++)
    if(n%i==0)
    return 0;
   return 1;
}
int main()
{
    int t=0,m;
    for(int i=2;i<=10005;i++)
        if(prime(i)==1)
          a[t++]=i*i;
    cin>>T;
    while(T--)
    {
        cin>>n;
        if(n==1) cout<<"4"<<endl;
        else {
      for(int i=0;i<t;i++)
        if(a[i]>=n)
         {m=i;
         break;}
        if( (n-a[m-1])<=(a[m]-n))
            cout<<a[m-1]<<endl;
        else cout<<a[m]<<endl;
        }
    }
    return 0;
}