计蒜客 等差数列

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

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

输入格式

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

输出格式

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

样例输入

5
7

样例输出

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

思路：暴力破解，三次循环嵌套

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

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