hdu 计算机学院大学生程序设计竞赛（2015’11）质方数

质方数

Time Limit: 2000/1000 MS (Java/Others)    MemoryLimit: 32768/32768 K (Java/Others)
Total Submission(s): 0    Accepted Submission(s): 0

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,tt;
    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)
         {tt=i;
         break;}
        if( (n-a[tt-1])<=(a[tt]-n))
            cout<<a[tt-1]<<endl;
        else cout<<a[tt]<<endl;
        }
    }
    return 0;
}