时限为5s
所以先把双平方数集合预处理出来,暴力搜索枚举即可,注意及时剪枝
输出答案,应该先按b排序再按a排序
/*
ID:xsy97051
LANG:C++
TASK:ariprog
*/
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int rl=0,a[20000],b[20000];
int n,m,num[30000],len=0;
bool vis[125001];
void init(int m)//预处理双平方数集合
{
for(int i=0;i<=125000;i++)
vis[i]=0;
for(int i=0;i<=m;i++)
for(int j=0;j<=m;j++)
vis[i*i+j*j]=1;
for(int i=0;i<=m*m*2;i++)
if(vis[i])
{
len++;
num[len]=i;
}
return;
}
int main()
{
freopen("ariprog.in","r",stdin);
freopen("ariprog.out","w",stdout);
cin>>n>>m;
init(m);
m=m*m*2;
for(int i=1;i<=len;i++)
for(int j=i+1;j<=len;j++)
{
if((2-n)*num[i]+num[j]*(n-1)>m) break;
bool okay=1;
for(int k=1;k<=n-2;k++)
if(!vis[num[j]+(num[j]-num[i])*k])
{
okay=0;
break;
}
if(okay)
{
rl++;
a[rl]=num[i];
b[rl]=num[j]-num[i];
}
}
if (rl==0)
{
cout<<"NONE"<<endl;
return 0;
}
for(int i=1;i<=rl;i++)//应该先按b排序再按a排序
for(int j=i+1;j<=rl;j++)
if ((b[i]>b[j]) || ( (b[i]==b[j]) && (a[i]>a[j])))
{
swap(a[i],a[j]);
swap(b[i],b[j]);
}
for(int i=1;i<=rl;i++)
cout<<a[i]<<" "<<b[i]<<endl;
return 0;
}