Arithmetic Progressions

http://train.usaco.org/usacoprob2?a=0fq0SrWwk8v&S=ariprog

题目大意：

输入:N，M

#include 
#include 
#include 
using namespace std;
int n, m, c[32000];
bool f[125001];
struct sequence
{
	int a;
	int b;
} ans[10005];
int main()
{
	ifstream fin("ariprog.in");
	ofstream fout("ariprog.out");
	fin >> n >> m;
	int t = 0, num = 0;
	for(int i = 0; i <= m; i++)
		for(int j = i; j <= m; j++)
			if(!f[i*i + j*j])
			c[t++] = i*i + j*j, f[i*i + j*j] = true;
	sort(c, c + t);
	for(int i = 1; i <= (c[t - 1] - c[0])/(n - 1); i++)			 //i：公差 b = i
	{
		for(int j = 0; c[j] + i*(n - 1) <= c[t - 1]; j++)		 //j:a的起始位置 a = c[j]
		{
			bool isok = true;
			for(int k = 1; k < n; k++)
			{
				if(!f[c[j] + i*k])
				{
					isok = false;
					break;
				}
			}
			if(isok)
			{
				ans[num].a = c[j];
				ans[num].b = i;
				num++;
			}
		}
	}
	if(num == 0) fout << "NONE" << endl;
	else for(int i = 0; i < num; i++) fout << ans[i].a << " " << ans[i].b << endl;
	fout.close();
	return 0;
}

在由p²+q²（0<=p，q<=M）组成的集合中找出能达到N项的等差数列，输出等差数列的第一项a，公差b；输出时先按b从小到大输出，再按a排序

这是搜索里的题，我也不知道怎么回事就让我写成一道水题了。。。。总结：搜索枚举时对所有的情况一定要不重不漏（除非剪枝）