杭电2015‘11校赛 1004 质方数

1004 质方数

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 3247 Accepted Submission(s): 1080

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;
}