质方数

质方数

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)


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

解题思路：打表标记非素数，即i*j的都不为素数，然后将遍历数组非标记的即为素数，进行平方放入数组，即为质方数，最后寻找比n第一个大的质方数，然后再用n的前一个质方数，这俩个数分别与n算差值，取小的即为最接近

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int a[10009],record[2000],num;
void primeSquare()
{
int i,j;
for(i=2;i<5001;i++)//i<5001是因为最大限制是10^4,而2*5000 = 10000，所以5001
{
for(j=2;i*j<10009;j++)//为了避免数据的遗漏，将>10^4的第一素数为10007
{
a[i*j] = 1;//标记1为不是素数，因为i*j这个数的因子最少为i和j，所以它肯定不是素数
}
}
for(i=2;i<10009;i++)
{
if(a[i] == 0)//没有就标记的即为素数
{
record[num++] = i*i;//计算质方数
}

}
}
int main()
{
primeSquare();
int t,n,i;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
if(n <= 4)
{
printf("4\n");
continue;
}
for(i=0;i<num;i++)
{
if(n < record[i])//找到第一个比n大的质方数
{
break;
}
}
//i = lower_bound(record,record+num,n) - reocrd;//STL查找函数
if(n - record[i-1] > record[i] - n)//判断n的前一个质方数和后一个哪个差值小，即输出哪个质方数
printf("%d\n",record[i]);
else
printf("%d\n",record[i-1]);
}
return 0;
}