东华OJ 基础题54 等差数列——难度难

问题描述 :

一个等差数列是一个能表示成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个等差数列。

输入范例 :

5 7

输出范例 :

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

完整C++代码

想到思路就简单，先标记双平方数，然后直接暴力遍历

#include <bits/stdc++.h>

using namespace std;

int main()
{
	int arr[100000] = {0};
	int n, m;
	cin >> n >> m;
	// 标记双平方的位置 
	for (int i = 0; i <= m; i++) {
		for (int j = 0; j <= m; j++) {
			arr[i * i + j * j] = 1;
		}
	}
	bool isFind = false;
	// 从b先开始，题目要求b先排序 
	for (int b = 1; b <= m * m; b++) {
		for (int a = 0; a <= m * m; a++) {
			int i;
			// 判断前n个是否为双平方 
			for (i = 0; i < n; i++) {
				if (arr[a + i * b] == 0) break;
			}
			if (i == n) {
				isFind = true;
				cout << a << " " << b << endl;
			}
		}
	}
	if (!isFind) cout << "NONE" << endl; 
	return 0;
}

😋欢迎大伙私信或者评论区交流讨论😋