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組測試例超時半秒，用方法一隻用方法二的一半時間，可見應該儘可能控制好進入循環的條件而不是先進入再檢測。

 

/*
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;
}