使用暴力搜索,枚举b,a的所有情况,进行检测,注意剪枝。首先是使用数组bool x[i]保存i值是否满足i==p*p+q*q这一条件,这一使用空间换时间的方法比使用检测函数快得多。第二个是注意b和a的取值范围,b是[1,2*m*m/(n-1)],a是[0,2*m*m-(n-1)*b]。控制取值范围方法一:在for循环第二个表达式中控制好进入循环条件;方法二:先进入for循环再求值检测,不满足要求就跳出循环。使用方法二时第8组测试例超时半秒,用方法一只用方法二的一半时间,可见应该尽量控制好进入循环的条件而不是先进入再检测。
/*
ID:jzzlee1
PROG:ariprog
LANG:C++
*/
#include<iostream>
#include<fstream>
#include<cstring>
#include<cctype>
#include<cstdio>
using namespace std;
ifstream fin("ariprog.in");
ofstream fout("ariprog.out");
int n,m,s[26],x[1300000];
int main()
{
int a,b,i,mm;
fin>>n>>m;
//cin>>n>>m;
for(int p=0;p<=m;p++)
for(int q=0;q<=m;q++)
x[p*p+q*q]=1;
mm=2*m*m;bool flag=1,sign=0;
for(b=1;b<=mm/(n-1);b++)
{
for(a=0;a<=mm-(n-1)*b;a++)
{
flag=1;
for(i=0;i<n;i++)
{
s[i]=a+i*b;
if(!x[s[i]])
{
flag=0;
break;
}
}
if(flag)
{
//cout<<a<<" "<<b<<endl;
fout<<a<<" "<<b<<endl;
sign=1;
}
}
}
if(!sign)
//cout<<"NONE"<<endl;
fout<<"NONE"<<endl;
return 0;
}