7-16 1.4.3 等差数列 (90分)

一个等差数列是一个能表示成 a, a+b, a+2b,…, a+nb (n=0,1,2,3,…) 在这个问题中 a 是一个非负的整数,b 是正整数. 写一个程序来找出在双平方数集合 S 中长度为 n 的等差数列. 双平方数集合是所有能表示成 p2+q2 的数的集合.

输入格式:
第一行: N(3<= N<=25),要找的等差数列的长度. 第二行: M(1<= M<=250),搜索双平方数的上界 0 <= p,q <= M.

输出格式:
如果没有找到数列,输出`NONE’. 如果找到了,输出一行或多行, 每行由于二个整数组成:a,b 这些行应该先按 b 排序再按 a 排序. 将不会有只多于 10,000 个等差数列.

输入样例1:
在这里给出一组输入。例如：

5
7

输出样例1:
在这里给出相应的输出。例如：

1 4
37 4
2 8
29 8
1 12
5 12
13 12
17 12
5 20
2 24

输入样例2:

14
10

输入样例2:

NONE

输入样例3:

21
200

输出样例3:

1217 84
2434 168
4868 336
6085 420
9736 672
10953 756
12170 840
12953 924
15821 1092

---

【思路】

---

【源代码】

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;

int main()
{
	int N, M;
	cin >> N >> M;
	vector<int> List(M + 1); // 根据 单平方数集合List， 生成双平方数集合Bis； 
	vector<int> Bis;
	for (int i = 0; i <= M; i++) // step1:先生成单平方数集合List; 
	{
		List[i] = (i * i);
	}

	for (int i = 0; i <= M; i++) // 行标； 
	{
		for (int j = i; j <= M; j++) // 列标； 
		{
			Bis.push_back(List[i] + List[j]); // step2:生成双平方数集合Bis； 
		}
	}
	
	set<int>s(Bis.begin(), Bis.end()); // step3:对双平方数集合去重并排序，保证后面的遍历从小到大开始；
	Bis.assign(s.begin(), s.end());

	vector<bool> F(Bis[Bis.size() - 1] + 1); // stpe4:判别x是否在双平方数集合内，若存在，则 F[x] = 1;
	for (int i = 0; i < Bis.size(); i++)
	{
		F[Bis[i]] = 1;
	}

	int sum = 0; // 记录（满足条件的等差数列）的次数；
	int a_0 = 0; // 等差数列的首项
	int a_n = 0; // 等差数列的某项；
	int b = 0;   // 等差数列的公差；
	int length = 0; // 记录当前数列的长度；
	for (int i = 1; i <= (Bis[Bis.size() - 1] / (N - 1)); i++) // 1 <= b < ((M*M+M*M)/N) ；
	{
		b = i; // 先得到b，保证先按b从小到大的顺序遍历，再到a；
		for (int j = 0; j < Bis.size(); j++) // 遍历双平方数集合；
		{
			length = 0; // 重置数列长度
			a_0 = Bis[j]; // 得到a[0]；
			for (int k = 1; k <= N; k++) // 判断长度为N的等差数列的项是否在F中； 
			{
				a_n = a_0 + (k - 1) * b; // 等差数列的项：a[n] = a[0]+ (n-1) * n;
				if (a_n > Bis[Bis.size() - 1]) // 注（i），a[n] > 双平方数集合中的最大数 ； 
				{
					break;
				}

				if (F[a_n])
				{
					length++;
					continue;
				}
				else // 此对a、b构成的等差数列中的 某一项a[n] 不在双平方数的集合中； 
				{ 
					break;
				}

				if (length == N) // 注（ii），此等差数列长度刚好够 N；
				{
					break;
				}
			}
			if (length == N)
			{
				cout << a_0 << " " << b << endl;
				sum++;
			}
		}
	}
	if (sum == 0)
	{
		cout << "NONE" << endl;
	}
	return 0;
}