USACO1.4 Arithmetic Progressions(ariprog)

        使用暴力搜索,枚举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;
}
相关文章
相关标签/搜索