等差数列

时限：4000ms 内存限制：10000K  总时限：10000ms

描述：

一个等差数列是一个能表示成a, a+b, a+2b,..., a+(n-1)b (n=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个等差数列

输入样例：

5
7

输出样例：

1 4
37 4
2 8
29 8
1 12
5 12
13 12
17 12
5 20
2 24
#include <iostream> #include <fstream> using namespace std; int n, m, maxNumber; bool flag = false; int start[10001], store[10001]; int count = 0; void Generate(bool check[]) { for (int i = 0; i <= m; i++) { for (int j = i; j <= m; j++) { check[i * i + j * j] = true; } } } void SearchLine(bool check[]) { for (int step = 1; step <= maxNumber / (n - 1); step++) { for (int begin = 0; begin <= maxNumber - (n - 1) * step; begin++) { for (int stepTimes = 0; stepTimes < n && begin + stepTimes * step <= maxNumber;stepTimes++) { if (!check[begin + stepTimes * step]) break; } if (stepTimes == n) { flag = true; start[count] = begin; store[count] = step; count++; } } } } int main(int argc, char *argv[]) { cin >> n >> m; maxNumber = m * m * 2; ofstream fout("file.txt"); bool *check = new bool[maxNumber + 1]; memset(check, 0x00, maxNumber + 1); Generate(check); SearchLine(check); if (!flag) cout << "NONE/n"; else for (int i = 0; i < count; i++) cout << start[i] << " " << store[i] << endl; return 0; }