使用暴力搜索,枚举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组测试例超时半秒,用方法一只用方法二的一半时间,可见应该尽可能控制好进入循环的条件而不是先进入再检测。ios
/* 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; }