计蒜客 难题题库 026 等差数列

•  869次 
•  5.29% 
•  2000ms 
•  65536K

一个等差数列是一个能表示成a, a+b, a+2b,..., a+nb (n=0,1,2,3,...)的数列。

在这个问题中a是一个非负的整数，b是正整数。写一个程序来找出在双平方数集合(双平方数集合是所有能表示成p2＋q2的数的集合)S中长度为n的等差数列。

输入包括两行，第一行为N要找的等差数列的长度（3≤ N≤25）。第二行是找到的双平方数p和q的上界M（0≤p,q≤M）。

输出一行或者多行，如果没有找到数列,输出NONE。否则输出一个整数对a b（这些行应该先按b排序再按a排序）

样例1

输入：

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<vector>
#include<algorithm>
using namespace std;

const int maxm = 250;
bool in[maxm * maxm + 1];   // 标志 i 是否为双平方数
int cnt[maxm * maxm + 1];   // 标志 1...i 共有多少个双平方数

int main(){
    int n, m;
    cin >> n >> m;
    int i, j;
    in[0] = 1;
    for(i = 0; i <= m; i++){
        for(j = 0; j <= m; j++){
            in[i * i + j * j] = 1;
        }
    }
    int count = 0;
    int max_i = m * m * 2;
    int a, b;
    vector<pair<int, int>> res;
    for(a = 0; a <= max_i; ++a){
        if(!in[a]){
            continue;
        }
        for(b = 1; a + (n - 1) * b <= max_i; ++b){
            for(i = 0; i < n; ++i){
                if(!in[a + i * b]){
                    break;
                }
            }
            if(i == n){
                res.push_back(make_pair(b, a));
            }
        }
    }
    if(res.empty()){
        cout << "NONE" << endl;
        return 0;
    }
    sort(res.begin(), res.end());
	n = res.size();
    for(i = 0; i < n; ++i){
        cout << res[i].second << " " << res[i].first << endl;
    }
}