PAT1078

#include<iostream>
#include<string.h>
#include<string>
#include<cmath>
#include<math.h>
using namespace std;
int size,number;
bool a[10100]={false};
int b[10100];




bool isprime(int  x)
{
if (x==1) return (false);
for(int i=2;i<=sqrt(1.0*x);i++)
{
if ((x%i)==0){return (false);}
}
return(true);
}


void deal(int x,int rank)
{
int pos=x%size;
if (!a[pos]){a[pos]=true;b[rank]=pos;return;}
else
{
for(int i=1;i<size;i++)
{
pos=(x+i*i)%size;
if (!a[pos]){a[pos]=true;b[rank]=pos;return;}
}
}
return;
}
int main()
{
memset(b,-1,sizeof(b));
cin>>size>>number;
for(int i=size;;i++)
{
if(isprime(i))
{
size=i;
break;
}

}
for(int i=0;i<number;i++)
{
int t;
cin>>t;
deal(t,i);
}

for(int i=0;i<number-1;i++)
{
if (b[i]!=-1) {cout<<b[i]<<" ";}
else{cout<<"-"<<" ";}
}
if (b[number-1]!=-1) {cout<<b[number-1];}
else{cout<<"-";}

system("pause");
return 0;

}

注意点：

1.1不是质数！

2.判断循环到size即可

3.sqrt的输入输出均为double，故要用1.0去乘