计蒜客等差数列

题目链接：点击打开链接

#include<stdio.h>
#include<string.h> 
int main()
{
int n,m,l,i,j,ans,k,temp,s;
int visit[200010],a[200010];
scanf("%d%d",&n,&m);
memset(visit,0,sizeof(visit));
for(i=0;i<=m;i++)
for(j=0;j<=m;j++)
visit[i*i+j*j]=1;
l=0;
for(i=0;i<=m*m*2;i++)
if(visit[i])
a[l++]=i;
s=m*m*2/(n-1);
temp=0; 
for(j=1;j<=s;j++)
{
for(i=0;a[i]+(n-1)*j<=m*m*2&&i<l;i++)
{
ans=1;
for(k=1;k<n;k++)
if(visit[a[i]+k*j]==0)
{
ans=0;
break;
}
if(ans)
{
printf("%d %d\n",a[i],j);
temp=1;
}
}
}
if(!temp)
printf("NONE\n");
return 0;
 }